mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 14:08:08 +01:00
110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
from models.tool_definitions import ToolsData
|
|
from typing import Optional
|
|
import json
|
|
|
|
|
|
def generate_genai_prompt(
|
|
tools_data: ToolsData, conversation_history: str, raw_json: Optional[str] = None
|
|
) -> str:
|
|
"""
|
|
Generates a concise prompt for producing or validating JSON instructions
|
|
with the provided tools and conversation history.
|
|
"""
|
|
prompt_lines = []
|
|
|
|
# Intro / Role
|
|
prompt_lines.append(
|
|
"You are an AI agent that helps fill required arguments for the tools described below. "
|
|
"You must respond with valid JSON ONLY, using the schema provided in the instructions."
|
|
)
|
|
|
|
# Main Conversation History
|
|
prompt_lines.append("=== Conversation History ===")
|
|
prompt_lines.append(
|
|
"This is the ongoing history to determine which tool and arguments to gather:"
|
|
)
|
|
prompt_lines.append("BEGIN CONVERSATION HISTORY")
|
|
prompt_lines.append(json.dumps(conversation_history, indent=2))
|
|
prompt_lines.append("END CONVERSATION HISTORY")
|
|
prompt_lines.append("")
|
|
|
|
# Example Conversation History (from tools_data)
|
|
if tools_data.example_conversation_history:
|
|
prompt_lines.append("=== Example Conversation With These Tools ===")
|
|
prompt_lines.append(
|
|
"Use this example to understand how tools are invoked and arguments are gathered."
|
|
)
|
|
prompt_lines.append("BEGIN EXAMPLE")
|
|
prompt_lines.append(tools_data.example_conversation_history)
|
|
prompt_lines.append("END EXAMPLE")
|
|
prompt_lines.append("")
|
|
|
|
# Tools Definitions
|
|
prompt_lines.append("=== Tools Definitions ===")
|
|
prompt_lines.append(f"There are {len(tools_data.tools)} available tools:")
|
|
prompt_lines.append(", ".join([t.name for t in tools_data.tools]))
|
|
prompt_lines.append(f"Goal: {tools_data.description}")
|
|
prompt_lines.append(
|
|
"Gather the necessary information for each tool in the sequence described above."
|
|
)
|
|
prompt_lines.append(
|
|
"Only ask for arguments listed below. Do not add extra arguments."
|
|
)
|
|
prompt_lines.append("")
|
|
for tool in tools_data.tools:
|
|
prompt_lines.append(f"Tool name: {tool.name}")
|
|
prompt_lines.append(f" Description: {tool.description}")
|
|
prompt_lines.append(" Required args:")
|
|
for arg in tool.arguments:
|
|
prompt_lines.append(f" - {arg.name} ({arg.type}): {arg.description}")
|
|
prompt_lines.append("")
|
|
prompt_lines.append(
|
|
"When all required args for a tool are known, you can propose next='confirm' to run it."
|
|
)
|
|
|
|
# JSON Format Instructions
|
|
prompt_lines.append("=== Instructions for JSON Generation ===")
|
|
prompt_lines.append(
|
|
"Your JSON format must be:\n"
|
|
"{\n"
|
|
' "response": "<plain text>",\n'
|
|
' "next": "<question|confirm|done>",\n'
|
|
' "tool": "<tool_name or null>",\n'
|
|
' "args": {\n'
|
|
' "<arg1>": "<value1 or null>",\n'
|
|
' "<arg2>": "<value2 or null>",\n'
|
|
" ...\n"
|
|
" }\n"
|
|
"}"
|
|
)
|
|
prompt_lines.append(
|
|
"1) If any required argument is missing, set next='question' and ask the user.\n"
|
|
"2) If all required arguments are known, set next='confirm' and specify the tool.\n"
|
|
" The user will confirm before the tool is run.\n"
|
|
"3) If no more tools are needed, set next='done' and tool=null.\n"
|
|
"4) response should be short and user-friendly."
|
|
)
|
|
|
|
# Validation Task (If raw_json is provided)
|
|
if raw_json is not None:
|
|
prompt_lines.append("")
|
|
prompt_lines.append("=== Validation Task ===")
|
|
prompt_lines.append("Validate and correct the following JSON if needed:")
|
|
prompt_lines.append(json.dumps(raw_json, indent=2))
|
|
prompt_lines.append("")
|
|
prompt_lines.append(
|
|
"Check syntax, 'tool' validity, 'args' completeness, "
|
|
"and set 'next' appropriately. Return ONLY corrected JSON."
|
|
)
|
|
|
|
# Prompt Start
|
|
prompt_lines.append("")
|
|
if raw_json is not None:
|
|
prompt_lines.append("Begin by validating the provided JSON if necessary.")
|
|
else:
|
|
prompt_lines.append(
|
|
"Begin by producing a valid JSON response for the next tool or question."
|
|
)
|
|
|
|
return "\n".join(prompt_lines)
|