pyagentkit 0.1.2__tar.gz → 0.1.5__tar.gz

{pyagentkit-0.1.2 → pyagentkit-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyagentkit
-Version: 0.1.2
+Version: 0.1.5
 Summary: Agent toolkit for Ollama
 Project-URL: Homepage, https://github.com/TarikEren/pyagentkit
 Project-URL: Issues, https://github.com/TarikEren/pyagentkit/issues
@@ -44,7 +44,10 @@ pip install pyagentkit
 ## Example
 
 ```python
-from ollama_agentkit import Agent, tool, AgentResponse
+from ollama_agentkit import Agent, tool, AgentResponse, configure_logging
+
+# For configuring agent logging
+configure_logging()
 
 # Custom output class to manipulate LLM output
 class CustomOutput(AgentResponse):
@@ -58,9 +61,7 @@ agent = Agent(
     response_model=CustomOutput
 )
 
-# Easy to define tooling with decorator `register_tool`
-@agent.register_tool
-def add_tool(n1: int, n2: int):
+def sum_tool(n1: int, n2: int):
     """Adds two numbers""" # All tooling must have a one-liner docstring that defines what the function does
     # Every tool must return a `ToolResult` object
     return ToolResult(
@@ -69,8 +70,12 @@ def add_tool(n1: int, n2: int):
         content=f"Result: {n1 + n2}"              # The part which the agent will read
     )
 
+agent.add_tool(sum_tool)
+
 # Messages get printed during the response handling
 result = agent.handle_response(prompt="What is 2 + 2")
+
+# Custom output class fields can be accessed as such
 print(result.sum_result)
 
 ```

{pyagentkit-0.1.2 → pyagentkit-0.1.5}/README.md

@@ -24,7 +24,10 @@ pip install pyagentkit
 ## Example
 
 ```python
-from ollama_agentkit import Agent, tool, AgentResponse
+from ollama_agentkit import Agent, tool, AgentResponse, configure_logging
+
+# For configuring agent logging
+configure_logging()
 
 # Custom output class to manipulate LLM output
 class CustomOutput(AgentResponse):
@@ -38,9 +41,7 @@ agent = Agent(
     response_model=CustomOutput
 )
 
-# Easy to define tooling with decorator `register_tool`
-@agent.register_tool
-def add_tool(n1: int, n2: int):
+def sum_tool(n1: int, n2: int):
     """Adds two numbers""" # All tooling must have a one-liner docstring that defines what the function does
     # Every tool must return a `ToolResult` object
     return ToolResult(
@@ -49,8 +50,12 @@ def add_tool(n1: int, n2: int):
         content=f"Result: {n1 + n2}"              # The part which the agent will read
     )
 
+agent.add_tool(sum_tool)
+
 # Messages get printed during the response handling
 result = agent.handle_response(prompt="What is 2 + 2")
+
+# Custom output class fields can be accessed as such
 print(result.sum_result)
 
 ```

{pyagentkit-0.1.2 → pyagentkit-0.1.5}/USAGE.md

@@ -1,12 +1,17 @@
 # How To Use Pyagentkit
 
+- Important: To configure agent logging, run `configure_logging()`
+
 ## Defining Agents
 
 - Defining a `base` agent is as easy as:
 
 ```python
 
-from pyagentkit import Agent
+from pyagentkit import Agent, configure_logging
+
+configure_logging() # For logging agents' logging
+configure_logging(level=logging.DEBUG) # For logging with debugging (Useful for detecting any issues with the agents' responses)
 
 agent = Agent(llm_name="<llm_name>")
 response = agent.handle_response(prompt="Hello, World!")

{pyagentkit-0.1.2 → pyagentkit-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pyagentkit"
-version = "0.1.2"
+version = "0.1.5"
 description = "Agent toolkit for Ollama"
 readme = "README.md"
 requires-python = ">=3.12"

{pyagentkit-0.1.2 → pyagentkit-0.1.5}/src/pyagentkit/agent.py

@@ -33,7 +33,7 @@ from .exceptions import (
 
 T = TypeVar("T", bound=AgentResponse)
 
-logger = logging.Logger("pyagentkit")
+logger = logging.getLogger("pyagentkit")
 
 
 class Agent(Generic[T]):
@@ -450,7 +450,7 @@ If task is done, generate `final` response and stop.""",
 
                 validated = self.response_model.model_validate_json(content)
 
-                logging.info("[%s]: %s", self.agent_name, validated.message)
+                logger.info("[%s]: %s", self.agent_name, validated.message)
                 self._validate_agent_logic(response=validated)
                 if validated.response.type == "final":
                     return validated