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

gemini_agent_framework-0.1.14/.github/workflows/deploy-docs.yml

@@ -0,0 +1 @@
+ 

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

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: gemini-agent-framework
-Version: 0.1.13
+Version: 0.1.14
 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
+Project-URL: Documentation, https://m7mdony.github.io/gemini-agent-framework
 Project-URL: Repository, https://github.com/m7mdony/gemini-agent-framework.git
 Project-URL: Issues, https://github.com/m7mdony/gemini-agent-framework/issues
 Author-email: Mohamed Baathman <mohamed.baathman2001@gmail.com>

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "gemini-agent-framework"
-version = "0.1.13"
+version = "0.1.14"
 description = "A framework for building agents that use Gemini's function calling capabilities"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -30,7 +30,7 @@ dependencies = [
 
 [project.urls]
 Homepage = "https://github.com/m7mdony/gemini-agent-framework"
-Documentation = "https://github.com/m7mdony/gemini-agent-framework/wiki"
+Documentation = "https://m7mdony.github.io/gemini-agent-framework"
 Repository = "https://github.com/m7mdony/gemini-agent-framework.git"
 Issues = "https://github.com/m7mdony/gemini-agent-framework/issues"
 

{gemini_agent_framework-0.1.13 → gemini_agent_framework-0.1.14}/src/gemini_agent/__init__.py

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

{gemini_agent_framework-0.1.13 → gemini_agent_framework-0.1.14}/src/gemini_agent/agent.py

@@ -4,9 +4,12 @@ from datetime import datetime
 from functools import wraps
 from typing import Any, Callable, Dict, List, Optional, Type, Union, Collection
 
+
 import requests
 from dotenv import load_dotenv
 
+
+
 load_dotenv()
 
 
@@ -237,14 +240,7 @@ class Agent:
             ]
         )
 
-        return """You are an AI assistant that can break down complex tasks into sequential steps using available tools.
-        When faced with a complex request:
-        1. Analyze the request to identify which tools can be used
-        2. Break down the request into sequential steps
-        3. For each step:
-           - Use the most appropriate tool
-           - Store the result for use in subsequent steps
-        4. Combine the results to provide a final answer
+        return """
         
         Available tools:
         {tools_list}
@@ -281,11 +277,26 @@ 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
@@ -328,21 +339,17 @@ class Agent:
             system_prompt = self._get_system_prompt() + system_prompt
 
         current_contents = conversation_history if conversation_history else []
-        if system_prompt and not current_contents:
-            current_contents.append({"role": "user", "parts": [{"text": system_prompt}]})
-            current_contents.append(
-                {
-                    "role": "model",
-                    "parts": [
-                        {
-                            "text": "I understand I should break down complex tasks into sequential steps using the available tools and variables."
-                        }
-                    ],
-                }
-            )
+        
+        # Add system instruction to payload
+        payload: Dict[str, Any] = {
+            "system_instruction": {
+                "parts": [{"text": system_prompt}]
+            },
+            "contents": current_contents
+        }
 
-        current_contents.append({"role": "user", "parts": [{"text": user_prompt}]})
-        payload: Dict[str, Any] = {"contents": current_contents}
+        # Add user prompt to contents
+        payload["contents"].append({"role": "user", "parts": [{"text": user_prompt}]})
 
         if self._registered_tools_json:
             payload["tools"] = [{"functionDeclarations": self._registered_tools_json}]
@@ -366,9 +373,9 @@ class Agent:
             }
             final_mime_type = "application/json"
             final_response_schema = response_structure
-
+     
         while True:
-
+           
             response_data = self._call_gemini_api(payload)
             if "error" in response_data:
                 print(
@@ -577,7 +584,6 @@ class Agent:
                                     )
                                     return final_text
                             return final_text
-
                 continue
 
             except (KeyError, IndexError) as e: