slipstream/discordbot
slipstream/discordbot
1
1
Fork 0
Code Issues Pull Requests 3 Actions Packages Projects Releases Wiki Activity

aa

Browse Source
This commit is contained in:
Slipstream 2025-04-28 16:45:52 -06:00
parent 4e9210b6b2
commit 067f4277c5
Signed by: slipstream
GPG Key ID: 13E498CE010AC6FD
2 changed files with 49 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
gurt/background.py
View File

@ -21,12 +21,17 @@ from .config import (
# Assuming analysis functions are moved
from .analysis import (
analyze_conversation_patterns, evolve_personality, identify_conversation_topics,
reflect_on_memories, decompose_goal_into_steps # Import goal decomposition
reflect_on_memories, decompose_goal_into_steps, # Import goal decomposition
proactively_create_goals # Import placeholder for proactive goal creation
)
if TYPE_CHECKING:
from .cog import GurtCog # For type hinting
# --- Tool Mapping Import ---
# Import the mapping to execute tools by name
from .tools import TOOL_MAPPING
# --- Background Task ---
async def background_processing_task(cog: 'GurtCog'):
@ -181,24 +186,55 @@ async def background_processing_task(cog: 'GurtCog'):
if tool_name:
print(f" - Attempting tool: {tool_name} with args: {tool_args}")
# --- TODO: Implement Tool Execution Logic ---
# 1. Find tool_func in TOOL_MAPPING
# 2. Execute tool_func(cog, **tool_args)
# 3. Handle success/failure of the tool call
# 4. Store tool result if needed for subsequent steps (requires modifying goal details/plan structure)
tool_success = False # Placeholder
tool_error = "Tool execution not yet implemented." # Placeholder
tool_func = TOOL_MAPPING.get(tool_name)
tool_result = None
tool_error = None
tool_success = False
if tool_func:
try:
# Ensure args are a dictionary, default to empty if None/missing
args_to_pass = tool_args if isinstance(tool_args, dict) else {}
print(f" - Executing: {tool_name}(cog, **{args_to_pass})")
start_time = time.monotonic()
tool_result = await tool_func(cog, **args_to_pass)
end_time = time.monotonic()
print(f" - Tool '{tool_name}' returned: {str(tool_result)[:200]}...") # Log truncated result
# Check result for success/error
if isinstance(tool_result, dict) and "error" in tool_result:
tool_error = tool_result["error"]
print(f" - Tool '{tool_name}' reported error: {tool_error}")
cog.tool_stats[tool_name]["failure"] += 1
else:
tool_success = True
print(f" - Tool '{tool_name}' executed successfully.")
cog.tool_stats[tool_name]["success"] += 1
# Record stats
cog.tool_stats[tool_name]["count"] += 1
cog.tool_stats[tool_name]["total_time"] += (end_time - start_time)
except Exception as exec_e:
tool_error = f"Exception during execution: {str(exec_e)}"
print(f" - Tool '{tool_name}' raised exception: {exec_e}")
traceback.print_exc()
cog.tool_stats[tool_name]["failure"] += 1
cog.tool_stats[tool_name]["count"] += 1 # Count failures too
else:
tool_error = f"Tool '{tool_name}' not found in TOOL_MAPPING."
print(f" - Error: {tool_error}")
# --- Handle Tool Outcome ---
if tool_success:
print(f" - Tool '{tool_name}' executed successfully.")
# Store result if needed (optional, requires plan structure modification)
# plan['step_results'][current_step_index] = tool_result
current_step_index += 1
else:
print(f" - Tool '{tool_name}' failed: {tool_error}")
goal_failed = True
plan['error_message'] = f"Failed at step {current_step_index + 1}: {tool_error}"
plan['error_message'] = f"Failed at step {current_step_index + 1} ({tool_name}): {tool_error}"
else:
# Step doesn't require a tool (e.g., internal reasoning/check)
print(" - No tool required for this step.")
print(" - No tool required for this step (internal check/reasoning).")
current_step_index += 1 # Assume non-tool steps succeed for now
# Check if goal completed

2
wheatley/api.py
View File

@ -57,7 +57,7 @@ from .config import (
PROJECT_ID, LOCATION, DEFAULT_MODEL, FALLBACK_MODEL,
API_TIMEOUT, API_RETRY_ATTEMPTS, API_RETRY_DELAY, TOOLS, RESPONSE_SCHEMA,
PROACTIVE_PLAN_SCHEMA, # Import the new schema
TAVILY_API_KEY, PISTON_API_URL, PISTON_API_KEY, BASELINE_PERSONALITY # Import other needed configs
TAVILY_API_KEY, PISTON_API_URL, PISTON_API_KEY # Import other needed configs
)
from .prompt import build_dynamic_system_prompt
from .context import gather_conversation_context, get_memory_context # Renamed functions