gemini-agent-framework 0.1.3__tar.gz → 0.1.5__tar.gz

{gemini_agent_framework-0.1.3 → gemini_agent_framework-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-agent-framework
-Version: 0.1.3
+Version: 0.1.5
 Summary: A framework for building agents that use Gemini's function calling capabilities
 Project-URL: Homepage, https://github.com/m7mdony/gemini-agent-framework
 Project-URL: Documentation, https://github.com/m7mdony/gemini-agent-framework#readme

{gemini_agent_framework-0.1.3 → gemini_agent_framework-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "gemini-agent-framework"
-version = "0.1.3"
+version = "0.1.5"
 description = "A framework for building agents that use Gemini's function calling capabilities"
 readme = "README.md"
 requires-python = ">=3.8"

gemini_agent_framework-0.1.5/src/gemini_agent/__init__.py

@@ -0,0 +1,4 @@
+from .agent import Agent
+
+__version__ = "0.1.5"
+__all__ = ["Agent"] 

{gemini_agent_framework-0.1.3 → gemini_agent_framework-0.1.5}/src/gemini_agent/agent.py

@@ -213,6 +213,7 @@ class Agent:
         - If a step requires multiple tools, execute them sequentially
         - If you're unsure about the next step, explain your reasoning
         - You can use both stored variables and values from the prompt
+        - You don't have the ability to set new varaibles during the conversation
         - When using stored variables, ALWAYS use the {{"variable": "variable_name"}} syntax
         """.format(
             tools_list="\n".join([f"- {name}: {desc}" for name, desc in 
@@ -321,8 +322,11 @@ class Agent:
             }
             final_mime_type = "application/json"
             final_response_schema = response_structure
-
+        counter = 0
         while True:
+            with open(f"payload_variable_{counter}.json", "w") as f:
+                json.dump(payload, f)
+            
             response_data = self._call_gemini_api(payload)
             if "error" in response_data:
                 print(f"API call failed: {response_data['error'].get('message', 'Unknown API error')}")
@@ -468,8 +472,9 @@ class Agent:
                                     print(f"Warning: Failed to parse structured output after formatting call: {e}. Returning intermediate text.")
                                     return final_text
                             return final_text
-
+                counter += 1
                 continue
+                
 
             except (KeyError, IndexError) as e:
                 print(f"Error parsing API response structure: {e}. Response: {response_data}")

gemini_agent_framework-0.1.3/src/gemini_agent/__init__.py

@@ -1,4 +0,0 @@
-from .agent import Agent
-
-__version__ = "0.1.1"
-__all__ = ["Agent"] 

gemini_agent_framework-0.1.3/tests/__init__.py

@@ -1,3 +0,0 @@
-"""
-Test package for the Gemini Agent Framework.
-""" 

gemini_agent_framework-0.1.3/tests/test_class_methods.py

@@ -1,67 +0,0 @@
-import os
-from dotenv import load_dotenv
-from gemini_agent import Agent
-
-load_dotenv()
-
-class Calculator:
-    def __init__(self):
-        self.memory = 0
-
-    @Agent.description("Multiplies a number by the stored memory value.")
-    @Agent.parameters({
-        'number': {'type': int, 'description': 'The number to multiply with memory'}
-    })
-    def multiply_with_memory(self, number: int) -> int:
-        result = self.memory * number
-        self.memory = result
-        return result
-
-    @Agent.description("Adds a number to the stored memory value.")
-    @Agent.parameters({
-        'number': {'type': int, 'description': 'The number to add to memory'}
-    })
-    def add_to_memory(self, number: int) -> int:
-        result = self.memory + number
-        self.memory = result
-        return result
-
-def test_class_methods():
-    # Create a calculator instance
-    calculator = Calculator()
-    
-    # Create an agent with the calculator methods
-    agent = Agent(
-        api_key=os.getenv("GEMINI_API_KEY"),
-        tools=[calculator.multiply_with_memory, calculator.add_to_memory]
-    )
-
-    # Test using class methods
-    response = agent.prompt(
-        "Multiply 5 with memory (starting at 0), then add 10 to the result",
-        response_structure={
-            "type": "object",
-            "properties": {
-                "used_functions": {
-                    "type": "array",
-                    "items": {
-                        "type": "object",
-                        "properties": {
-                            "function_name": {"type": "string"},
-                            "parameters": {
-                                "type": "object",
-                                "properties": {
-                                    "number": {"type": "integer"}
-                                }
-                            }
-                        }
-                    }
-                },
-                "answer": {"type": "string"}
-            }
-        }
-    )
-    print(response)
-
-if __name__ == "__main__":
-    test_class_methods() 

gemini_agent_framework-0.1.3/tests/test_variables.py

@@ -1,133 +0,0 @@
-from agent import Agent
-import os
-from dotenv import load_dotenv
-import json
-from bs4 import BeautifulSoup
-
-load_dotenv()
-
-class HTMLAnalyzer:
-    @Agent.description("Count the number of input tags in an HTML page")
-    @Agent.parameters({
-        'html_content': {'type': str, 'description': 'The HTML content to analyze'}
-    })
-    def count_inputs(self, html_content: str) -> dict:
-        soup = BeautifulSoup(html_content, 'html.parser')
-        input_tags = soup.find_all('input')
-        return {
-            "count": len(input_tags),
-            "input_types": [tag.get('type', 'unknown') for tag in input_tags]
-        }
-
-# Create HTML pages
-home_page = """
-<!DOCTYPE html>
-<html>
-<head>
-    <title>Home Page</title>
-</head>
-<body>
-    <header>
-        <nav>
-            <a href="/">Home</a>
-            <a href="/about">About</a>
-            <a href="/contact">Contact</a>
-        </nav>
-    </header>
-    <main>
-        <h1>Welcome to Our Website</h1>
-        <p>This is a simple home page with no input fields.</p>
-    </main>
-    <footer>
-        <p>&copy; 2024 Our Website</p>
-    </footer>
-</body>
-</html>
-"""
-
-login_page = """
-<!DOCTYPE html>
-<html>
-<head>
-    <title>Login Page</title>
-</head>
-<body>
-    <header>
-        <nav>
-            <a href="/">Home</a>
-            <a href="/login">Login</a>
-        </nav>
-    </header>
-    <main>
-        <h1>Login</h1>
-        <form action="/login" method="POST">
-            <div>
-                <label for="username">Username:</label>
-                <input type="text" id="username" name="username" required>
-            </div>
-            <div>
-                <label for="password">Password:</label>
-                <input type="password" id="password" name="password" required>
-            </div>
-            <div>
-                <input type="checkbox" id="remember" name="remember">
-                <label for="remember">Remember me</label>
-            </div>
-            <button type="submit">Login</button>
-        </form>
-    </main>
-    <footer>
-        <p>&copy; 2024 Our Website</p>
-    </footer>
-</body>
-</html>
-"""
-
-# Create the analyzer instance
-html_analyzer = HTMLAnalyzer()
-
-# Create the agent with our tool
-agent = Agent(
-    api_key=os.getenv("GEMINI_API_KEY"),
-    tools=[html_analyzer.count_inputs]
-)
-
-# Store the HTML pages as variables
-agent.set_variable(
-    name="home_page",
-    value=home_page,
-    description="The HTML content of the home page",
-    type_hint=str
-)
-
-
-
-# Example 1: Count inputs in home page
-print("\nExample 1: Count inputs in home page")
-response = agent.prompt(
-    "How many input tags are in the home page?",
-    response_structure={
-        "type": "object",
-        "properties": {
-            "count": {"type": "integer"},
-            "input_types": {"type": "array", "items": {"type": "string"}}
-        },
-        "required": ["count", "input_types"]
-    }
-)
-print(json.dumps(response, indent=2))
-
-# Example 2: Count inputs in login page
-print("\nExample 2: Count inputs in login page")
-response = agent.prompt(
-    "How many input tags are in this page  "+ login_page,
-    response_structure={
-        "type": "object",
-        "properties": {
-            "count": {"type": "integer"},
-            "input_types": {"type": "array", "items": {"type": "string"}}
-        },
-        "required": ["count", "input_types"]
-    }
-)
-print(json.dumps(response, indent=2))