mirror/temporal-ai-agent
1
0
Fork 0
mirror of https://github.com/temporal-community/temporal-ai-agent.git synced 2026-03-15 14:08:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

google gemini 2.0 exp by default

Browse Source
This commit is contained in:
Steve Androulakis
2025-01-24 15:59:14 -08:00
parent 20fdf935ab
commit c56f45bb86
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
activities/tool_activities.py
View File

@@ -105,7 +105,7 @@ class ToolActivities:
genai.configure(api_key=api_key) genai.configure(api_key=api_key)
model = genai.GenerativeModel( model = genai.GenerativeModel(
"models/gemini-1.5-flash", "models/gemini-2.0-flash-exp",
system_instruction=input.context_instructions, system_instruction=input.context_instructions,
) )
response = model.generate_content(input.prompt) response = model.generate_content(input.prompt)

1
workflows/tool_workflow.py
View File

@@ -60,6 +60,7 @@ class ToolWorkflow:
self.prompt_queue.append( self.prompt_queue.append(
f"### The '{current_tool}' tool completed successfully with {dynamic_result}. " f"### The '{current_tool}' tool completed successfully with {dynamic_result}. "
"INSTRUCTIONS: Parse this tool result as plain text, and use the system prompt containing the list of tools in sequence and the conversation history (and previous tool_results) to figure out next steps, if any. " "INSTRUCTIONS: Parse this tool result as plain text, and use the system prompt containing the list of tools in sequence and the conversation history (and previous tool_results) to figure out next steps, if any. "
"You will need to use the tool_results to auto-fill arguments for subsequent tools and also to figure out if all tools have been run."
'{"next": "<question|confirm|done>", "tool": "<tool_name or null>", "args": {"<arg1>": "<value1 or null>", "<arg2>": "<value2 or null>}, "response": "<plain text>"}' '{"next": "<question|confirm|done>", "tool": "<tool_name or null>", "args": {"<arg1>": "<value1 or null>", "<arg2>": "<value2 or null>}, "response": "<plain text>"}'
"ONLY return those json keys (next, tool, args, response), nothing else." "ONLY return those json keys (next, tool, args, response), nothing else."
'Next should only be "done" if all tools have been run (use the system prompt to figure that out).' 'Next should only be "done" if all tools have been run (use the system prompt to figure that out).'