gemini-agent-framework 0.1.4__tar.gz → 0.1.6__tar.gz

{gemini_agent_framework-0.1.4 → gemini_agent_framework-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-agent-framework
-Version: 0.1.4
+Version: 0.1.6
 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.4 → gemini_agent_framework-0.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "gemini-agent-framework"
-version = "0.1.4"
+version = "0.1.6"
 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.6/src/gemini_agent/__init__.py

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

{gemini_agent_framework-0.1.4 → gemini_agent_framework-0.1.6}/src/gemini_agent/agent.py

@@ -213,7 +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
+        - You don't have the ability to set new varaibles during the conversation. and if you need to use them thhen you will need to use its actual value.
         - When using stored variables, ALWAYS use the {{"variable": "variable_name"}} syntax
         """.format(
             tools_list="\n".join([f"- {name}: {desc}" for name, desc in 
@@ -322,9 +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')}")
@@ -470,6 +472,7 @@ 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
                 
 

gemini_agent_framework-0.1.6/tests/test_agent.py

@@ -0,0 +1,76 @@
+from gemini_agent import Agent
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+# Load multiple API keys
+api_keys = os.getenv("GEMINI_APIs").split(",")
+current_key_idx = 0
+
+def get_current_api_key():
+    return api_keys[current_key_idx]
+
+def switch_to_next_api_key():
+    global current_key_idx
+    current_key_idx = (current_key_idx + 1) % len(api_keys)
+    print(f"🔄 [DEBUG] Switching to API key index {current_key_idx}: {api_keys[current_key_idx][:10]}...")
+
+# Define your tools
+@Agent.description("Multiplies two numbers.")
+@Agent.parameters({
+    'a': {'type': int, 'description': 'The first number'},
+    'b': {'type': int, 'description': 'The second number'}
+})
+def multiply(a: int, b: int) -> int:
+    return a * b
+
+@Agent.description("Adds two numbers.")
+@Agent.parameters({
+    'a': {'type': int, 'description': 'The first number'},
+    'b': {'type': int, 'description': 'The second number'}
+})
+def add(a: int, b: int) -> int:
+    return a + b
+
+# Create agent with the first API key
+agent = Agent(api_key=get_current_api_key(), tools=[multiply, add])
+
+# Define a wrapper to rotate API key AFTER the call
+def agent_prompt_with_key_rotation(agent,*args, **kwargs):
+    # global agent
+    print(f"🚀 [DEBUG] Using API key index {current_key_idx}: {agent.api_key[:10]}...")
+    response = agent.prompt(*args, **kwargs)
+    switch_to_next_api_key()
+    agent.api_key = get_current_api_key()
+    return response
+
+# Use the agent
+response = agent_prompt_with_key_rotation(agent,
+    user_prompt="multiply 3 and 7 then add 5 to the result",
+    system_prompt="You are a helpful assistant. Give your response always with ❤️ at the start of the line. In your response you should mention the function you used.",
+    response_structure={
+        "type": "object",
+        "properties": {
+            "used_functions": {
+                "type": "array",
+                "items": {
+                    "type": "object",
+                    "properties": {
+                        "function_name": {"type": "string"},
+                        "parameters": {
+                            "type": "object",
+                            "properties": {
+                                "a": {"type": "integer"},
+                                "b": {"type": "integer"}
+                            }
+                        }
+                    }
+                }
+            },
+            "answer": {"type": "string"}
+        }
+    }
+)
+
+print(response)

gemini_agent_framework-0.1.4/src/gemini_agent/__init__.py

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

gemini_agent_framework-0.1.4/tests/test_agent.py

@@ -1,66 +0,0 @@
-import os
-from dotenv import load_dotenv
-from framework_agent import Agent
-
-load_dotenv()
-
-def test_basic_operations():
-    # Define some basic math operations
-    @Agent.description("Multiplies two numbers.")
-    @Agent.parameters({
-        'a': {'type': int, 'description': 'The first number'},
-        'b': {'type': int, 'description': 'The second number'}
-    })
-    def multiply(a: int, b: int) -> int:
-        return a * b
-
-    @Agent.description("Adds two numbers.")
-    @Agent.parameters({
-        'a': {'type': int, 'description': 'The first number'},
-        'b': {'type': int, 'description': 'The second number'}
-    })
-    def add(a: int, b: int) -> int:
-        return a + b
-
-    # Create an agent with the math tools
-    agent = Agent(
-        api_key=os.getenv("GEMINI_API_KEY"),
-        tools=[multiply, add]
-    )
-
-    # Test a simple multiplication
-    response = agent.prompt("Multiply 3 and 7")
-    print(f"Multiplication result: {response}")  # Should be 21
-
-    # Use the agent
-    response = agent.prompt(user_prompt="multiply 3 and 7 then add 5 to the result" ,
-    system_prompt="You are a helpful assistant give your response always with ❤️ at the start of the line. in your response you should mention the function you used." ,
-    response_structure=
-    {
-        "type": "object",
-        "properties": {
-            "used_functions": {
-                "type": "array",
-                "items": {
-                    "type": "object",
-                    "properties": {
-                        "function_name": {"type": "string"},
-                        "parameters": {
-                            "type": "object",
-                            "properties": {
-                                "a": {"type": "integer"},
-                                "b": {"type": "integer"}
-                            }
-                        }
-                    }
-                }
-            },
-            "answer": {"type": "string"}
-            
-        }
-    }
-    )
-    print(response)  # Should output 21
-
-if __name__ == "__main__":
-    test_basic_operations()