feat(prompts): add different types of prompts, #6 (c4e929a1) · Commits · OOP / code / ai2

ai_agent/prompts/init.py

0 → 100644

+7 −0

Original line number	Diff line number	Diff line
		from ai_agent.prompts.memory import CONVERSATION_MEMORY_PROMPT
		from ai_agent.prompts.tool_agent import TOOL_USING_ASSISTANT_PROMPT

		__all__ = [
		"CONVERSATION_MEMORY_PROMPT",
		"TOOL_USING_ASSISTANT_PROMPT",
		]

0 → 100644

+5 −0

Original line number	Diff line number	Diff line
		CONVERSATION_MEMORY_PROMPT = """
		You are an intelligent assistant that uses available tools to respond to user requests.
		1. Identify the appropriate tool.
		2. Summarize the tool's result and be as concise as possible with the less
		"""

0 → 100644

+6 −0

Original line number	Diff line number	Diff line
		TOOL_USING_ASSISTANT_PROMPT = """
		You are an intelligent assistant that uses available tools to respond to user requests.
		1. Identify the appropriate tool.
		2. Summarize the tool's result naturally.
		3. Offer helpful next steps.
		"""

+2 −8

Original line number	Diff line number	Diff line
		import os
		import logging
		from dotenv import load_dotenv
		from ai_agent.prompts import CONVERSATION_MEMORY_PROMPT
		from langchain_groq import ChatGroq
		from mcp_use import MCPAgent, MCPClient # type: ignore[import-untyped]

		@@ -43,17 +44,10 @@ async def create_groq_agent():
		api_key=groq_api_key,
		)

		PROMPT = """
		You are an intelligent assistant that uses available tools to respond to user requests.
		1. Identify the appropriate tool.
		2. Summarize the tool's result naturally.
		3. Offer helpful next steps.
		"""

		agent = MCPAgent(
		client=client,
		max_steps=10,
		system_prompt=PROMPT,
		system_prompt=CONVERSATION_MEMORY_PROMPT,
		llm=llm,
		verbose=True,
		)