Context injection

cookbook/agents/24_agent_with_context.py

@@ -5,14 +5,6 @@
 
 
 def get_top_hackernews_stories(num_stories: int = 5) -> str:
-    """Use this function to get top stories from Hacker News.
-
-    Args:
-        num_stories (int): Number of stories to return. Defaults to 10.
-
-    Returns:
-        str: JSON string of top stories.
-    """
     # Fetch top story IDs
     response = httpx.get("https://hacker-news.firebaseio.com/v0/topstories.json")
     story_ids = response.json()
@@ -29,7 +21,6 @@ def get_top_hackernews_stories(num_stories: int = 5) -> str:
 
 
 agent = Agent(
-    tools=[get_top_hackernews_stories],
     context={
         "name": "John Doe",
         "top_stories": lambda: get_top_hackernews_stories(3),

cookbook/agents/25_system_prompt_via_function.py

@@ -0,0 +1,14 @@
+from phi.agent import Agent
+
+
+def get_system_prompt(agent: Agent) -> str:
+    return f"You are {agent.name}!"
+
+
+agent = Agent(
+    name="AgentX",
+    system_prompt=get_system_prompt,
+    markdown=True,
+    show_tool_calls=True,
+)
+agent.print_response("Who are you?", stream=True)

cookbook/agents/26_instructions_via_function.py

@@ -0,0 +1,16 @@
+from typing import List
+
+from phi.agent import Agent
+
+
+def get_instructions(agent: Agent) -> List[str]:
+    return ["Talk in haiku's!", "Use poetry to answer questions."]
+
+
+agent = Agent(
+    name="AgentX",
+    instructions=get_instructions,
+    markdown=True,
+    show_tool_calls=True,
+)
+agent.print_response("Who are you?", stream=True)

phi/agent/agent.py

@@ -494,7 +494,6 @@ def update_model(self) -> None:
 
     def _resolve_context(self) -> None:
         from inspect import signature
-        from functools import partial
 
         logger.debug("Resolving context")
         if self.context is not None:
@@ -504,8 +503,7 @@ def _resolve_context(self) -> None:
                         sig = signature(ctx_value)
                         resolved_ctx_value = None
                         if "agent" in sig.parameters:
-                            ctx_value_partial = partial(ctx_value, agent=self)
-                            resolved_ctx_value = ctx_value_partial()
+                            resolved_ctx_value = ctx_value(agent=self)
                         else:
                             resolved_ctx_value = ctx_value()
                         if resolved_ctx_value is not None:

phi/model/google/gemini.py

@@ -183,7 +183,10 @@ def _format_functions(self, params: Dict[str, Any]) -> Dict[str, Any]:
         return formatted_params
 
     def add_tool(
-        self, tool: Union["Tool", "Toolkit", Callable, dict, "Function"], strict: bool = False, agent: Optional[Any] = None
+        self,
+        tool: Union["Tool", "Toolkit", Callable, dict, "Function"],
+        strict: bool = False,
+        agent: Optional[Any] = None,
     ) -> None:
         """
         Adds tools to the model.