gemini-agent-framework 0.1.12__tar.gz → 0.1.13__tar.gz

{gemini_agent_framework-0.1.12 → gemini_agent_framework-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-agent-framework
-Version: 0.1.12
+Version: 0.1.13
 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/wiki
@@ -118,11 +118,11 @@ response = agent.prompt(
 
 ## Documentation
 
-For detailed documentation, please visit our [documentation site](https://github.com/m7mdony/gemini-agent-framework/wiki).
+For detailed documentation, please visit our [documentation site](https://github.com/m7mdony/gemini-agent-framework).
 
 ### Key Topics
 
-- [API Reference](https://m7mdony.github.io/gemini-agent-framework/api_refrences/)
+- [API Reference](https://m7mdony.github.io/gemini-agent-framework/api_reference/)
 - [Tutorials](https://m7mdony.github.io/gemini-agent-framework/tutorials/)
 - [Best Practices](https://m7mdony.github.io/gemini-agent-framework/best_practices/)
 - [Architecture Overview](https://m7mdony.github.io/gemini-agent-framework/architecture/)

{gemini_agent_framework-0.1.12 → gemini_agent_framework-0.1.13}/README.md

@@ -86,11 +86,11 @@ response = agent.prompt(
 
 ## Documentation
 
-For detailed documentation, please visit our [documentation site](https://github.com/m7mdony/gemini-agent-framework/wiki).
+For detailed documentation, please visit our [documentation site](https://github.com/m7mdony/gemini-agent-framework).
 
 ### Key Topics
 
-- [API Reference](https://m7mdony.github.io/gemini-agent-framework/api_refrences/)
+- [API Reference](https://m7mdony.github.io/gemini-agent-framework/api_reference/)
 - [Tutorials](https://m7mdony.github.io/gemini-agent-framework/tutorials/)
 - [Best Practices](https://m7mdony.github.io/gemini-agent-framework/best_practices/)
 - [Architecture Overview](https://m7mdony.github.io/gemini-agent-framework/architecture/)

{gemini_agent_framework-0.1.12 → gemini_agent_framework-0.1.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "gemini-agent-framework"
-version = "0.1.12"
+version = "0.1.13"
 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.12 → gemini_agent_framework-0.1.13}/src/gemini_agent/__init__.py

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

gemini_agent_framework-0.1.13/tests/__init__.py

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

gemini_agent_framework-0.1.13/tests/test_class_methods.py

@@ -0,0 +1,67 @@
+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.13/tests/test_variables.py

@@ -0,0 +1,133 @@
+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))