gemini-agent-framework 0.2.1__tar.gz → 0.2.3__tar.gz

{gemini_agent_framework-0.2.1 → gemini_agent_framework-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-agent-framework
-Version: 0.2.1
+Version: 0.2.3
 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://m7mdony.github.io/gemini-agent-framework

{gemini_agent_framework-0.2.1 → gemini_agent_framework-0.2.3}/pyproject.toml

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

{gemini_agent_framework-0.2.1 → gemini_agent_framework-0.2.3}/src/gemini_agent/__init__.py

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

{gemini_agent_framework-0.2.1 → gemini_agent_framework-0.2.3}/src/gemini_agent/agent.py

@@ -274,26 +274,20 @@ class Agent:
         """Substitutes variable references in arguments with their actual values."""
         result = {}
         for key, value in args.items():
-            print("substituting variables", key, value)
             if isinstance(value, str) and value.startswith("$"):
-                print("is string and starts with $")
                 # Handle $ prefixed variables
                 var_name = value[1:]
                 if var_name in self._stored_variables:
-                    print("substituted")
 
                     result[key] = self._stored_variables[var_name]["value"]
                 else:
                     result[key] = value
             elif isinstance(value, dict) and "variable" in value:
-                print("is dict and has variable")
                 # Handle dictionary-style variable references
                 var_name = value["variable"]
                 if var_name in self._stored_variables:
-                    print("substituted")
                     result[key] = self._stored_variables[var_name]["value"]
                 else:
-                    print("substituted")
                     result[key] = value
             else:
                 result[key] = value
@@ -320,7 +314,6 @@ class Agent:
     
     def _log_json(self, json_data: Dict[str, Any], file_name: str, debug_scope: Optional[str] = None) -> None:
         """Logs the JSON data to a file."""
-        print("in log json")
         if "json" not in debug_scope:
             return
         with open(file_name, "w") as f:
@@ -394,7 +387,6 @@ class Agent:
             self._log_json(payload, f"payload_{count}.json", debug_scope)
             count += 1
             response_data = self._call_gemini_api(payload, debug_scope)
-            print("response data " , response_data)
             if "error" in response_data:
                 self._log_text(
                     f"API call failed: {response_data['error'].get('message', 'Unknown API error')}"

gemini_agent_framework-0.2.3/tests/formatting_payload_2.json

@@ -0,0 +1,82 @@
+{
+    "contents": [
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "text": "How many input tags are in the home page?"
+                }
+            ]
+        },
+        {
+            "role": "model",
+            "parts": [
+                {
+                    "functionCall": {
+                        "name": "count_inputs",
+                        "args": {
+                            "html_content": {
+                                "variable": "home_page"
+                            }
+                        }
+                    }
+                }
+            ]
+        },
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "text": "the return value of the function stored in the variable result_0"
+                }
+            ]
+        },
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "functionResponse": {
+                        "name": "count_inputs",
+                        "response": {
+                            "content": {
+                                "count": 0,
+                                "input_types": []
+                            },
+                            "key": "result_0",
+                            "content_type": "dict"
+                        }
+                    }
+                }
+            ]
+        },
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "text": "Based on our conversation above, please format the following information according to the requested JSON structure:\n\nThere are 0 input tags in the home page."
+                }
+            ]
+        }
+    ],
+    "generationConfig": {
+        "response_mime_type": "application/json",
+        "response_schema": {
+            "type": "object",
+            "properties": {
+                "count": {
+                    "type": "integer"
+                },
+                "input_types": {
+                    "type": "array",
+                    "items": {
+                        "type": "string"
+                    }
+                }
+            },
+            "required": [
+                "count",
+                "input_types"
+            ]
+        }
+    }
+}

gemini_agent_framework-0.2.3/tests/payload_0.json

@@ -0,0 +1,49 @@
+{
+    "system_instruction": {
+        "parts": [
+            {
+                "text": "You are a tool that counts the number of input tags in an HTML page."
+            },
+            {
+                "text": "\n\n        Available variables:\n        - home_page: The HTML content of the home page (Type: <class 'str'>)\n        \n        IMPORTANT - Variable Usage:\n        When you need to use a stored variable in a function call, you MUST use the following syntax:\n        - For function arguments: {\"variable\": \"variable_name\"}\n        - For example, if you want to use the 'current_user' variable in a function call:\n          {\"user_id\": {\"variable\": \"current_user\"}}\n        \n        Remember:\n        - Always perform one operation at a time\n        - Use intermediate results from previous steps\n        - If a step requires multiple tools, execute them sequentially\n        - If you're unsure about the next step, explain your reasoning\n        - You can use both stored variables and values from the prompt\n        - When using stored variables, ALWAYS use the {\"variable\": \"variable_name\"} syntax\n        "
+            }
+        ]
+    },
+    "contents": [
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "text": "How many input tags are in the home page?"
+                }
+            ]
+        }
+    ],
+    "tools": [
+        {
+            "functionDeclarations": [
+                {
+                    "name": "count_inputs",
+                    "description": "Count the number of input tags in an HTML page",
+                    "parameters": {
+                        "type": "OBJECT",
+                        "properties": {
+                            "html_content": {
+                                "type": "STRING",
+                                "description": "The HTML content to analyze"
+                            }
+                        },
+                        "required": [
+                            "html_content"
+                        ]
+                    }
+                }
+            ]
+        }
+    ],
+    "toolConfig": {
+        "functionCallingConfig": {
+            "mode": "AUTO"
+        }
+    }
+}

gemini_agent_framework-0.2.3/tests/payload_1.json

@@ -0,0 +1,90 @@
+{
+    "system_instruction": {
+        "parts": [
+            {
+                "text": "You are a tool that counts the number of input tags in an HTML page."
+            },
+            {
+                "text": "\n\n        Available variables:\n        - home_page: The HTML content of the home page (Type: <class 'str'>)\n        \n        IMPORTANT - Variable Usage:\n        When you need to use a stored variable in a function call, you MUST use the following syntax:\n        - For function arguments: {\"variable\": \"variable_name\"}\n        - For example, if you want to use the 'current_user' variable in a function call:\n          {\"user_id\": {\"variable\": \"current_user\"}}\n        \n        Remember:\n        - Always perform one operation at a time\n        - Use intermediate results from previous steps\n        - If a step requires multiple tools, execute them sequentially\n        - If you're unsure about the next step, explain your reasoning\n        - You can use both stored variables and values from the prompt\n        - When using stored variables, ALWAYS use the {\"variable\": \"variable_name\"} syntax\n        "
+            }
+        ]
+    },
+    "contents": [
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "text": "How many input tags are in the home page?"
+                }
+            ]
+        },
+        {
+            "role": "model",
+            "parts": [
+                {
+                    "functionCall": {
+                        "name": "count_inputs",
+                        "args": {
+                            "html_content": {
+                                "variable": "home_page"
+                            }
+                        }
+                    }
+                }
+            ]
+        },
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "text": "the return value of the function stored in the variable result_0"
+                }
+            ]
+        },
+        {
+            "role": "user",
+            "parts": [
+                {
+                    "functionResponse": {
+                        "name": "count_inputs",
+                        "response": {
+                            "content": {
+                                "count": 0,
+                                "input_types": []
+                            },
+                            "key": "result_0",
+                            "content_type": "dict"
+                        }
+                    }
+                }
+            ]
+        }
+    ],
+    "tools": [
+        {
+            "functionDeclarations": [
+                {
+                    "name": "count_inputs",
+                    "description": "Count the number of input tags in an HTML page",
+                    "parameters": {
+                        "type": "OBJECT",
+                        "properties": {
+                            "html_content": {
+                                "type": "STRING",
+                                "description": "The HTML content to analyze"
+                            }
+                        },
+                        "required": [
+                            "html_content"
+                        ]
+                    }
+                }
+            ]
+        }
+    ],
+    "toolConfig": {
+        "functionCallingConfig": {
+            "mode": "AUTO"
+        }
+    }
+}

{gemini_agent_framework-0.2.1 → gemini_agent_framework-0.2.3}/tests/test_variables.py

@@ -1,4 +1,4 @@
-from agent import Agent
+from gemini_agent import Agent
 import os
 from dotenv import load_dotenv
 import json
@@ -105,7 +105,10 @@ agent.set_variable(
 # 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?",
+
+    user_prompt="How many input tags are in the home page?",
+    system_prompt="You are a tool that counts the number of input tags in an HTML page.",
+    debug_scope=["json"],
     response_structure={
         "type": "object",
         "properties": {