gemini-agent-framework 0.1.13__py3-none-any.whl → 0.1.15__py3-none-any.whl

gemini_agent/__init__.py

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

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,18 +240,8 @@ 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
-        
-        Available tools:
-        {tools_list}
-        
+        return """
+
         Available variables:
         {variables_list}
         
@@ -281,11 +274,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
@@ -322,27 +330,20 @@ class Agent:
         """
         self._intermediate_results = {}
 
-        if not system_prompt:
-            system_prompt = self._get_system_prompt()
-        else:
-            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 if system_prompt else ""},{"text": self._get_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 +367,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 +578,6 @@ class Agent:
                                     )
                                     return final_text
                             return final_text
-
                 continue
 
             except (KeyError, IndexError) as e:

{gemini_agent_framework-0.1.13.dist-info → gemini_agent_framework-0.1.15.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: gemini-agent-framework
-Version: 0.1.13
+Version: 0.1.15
 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.15.dist-info/RECORD

@@ -0,0 +1,6 @@
+gemini_agent/__init__.py,sha256=LFX-2Emo9kXG9rmleXchzu7GyCiwMpXLoxGETOEeKi8,69
+gemini_agent/agent.py,sha256=gdtr6GDKZJxq6_rgbDxvdkZmMB75vUcMop1eBrQ1adY,26527
+gemini_agent_framework-0.1.15.dist-info/METADATA,sha256=T9RiY-0KCOFbHkFazezxcD6-2NQdEwJPeirO_96OU64,5033
+gemini_agent_framework-0.1.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+gemini_agent_framework-0.1.15.dist-info/licenses/LICENSE,sha256=NHp9eKeWbB-Fp6Ff24BXSemJOa6UEZa9IAp9U7O9oBM,1073
+gemini_agent_framework-0.1.15.dist-info/RECORD,,

gemini_agent_framework-0.1.13.dist-info/RECORD

@@ -1,6 +0,0 @@
-gemini_agent/__init__.py,sha256=VJDQ1Ve3Y3wf_xjRODOnr2gYm0bAg04sHA_i8HXGudc,69
-gemini_agent/agent.py,sha256=eY9Vjbew4N1dbY31oK37UVRb4a9w-StUUjCuAYCF8N8,26740
-gemini_agent_framework-0.1.13.dist-info/METADATA,sha256=Y_h_pZhD05QrRdYk2WrD0Xr2EeBjPun2JquhI8vIyCE,5039
-gemini_agent_framework-0.1.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-gemini_agent_framework-0.1.13.dist-info/licenses/LICENSE,sha256=NHp9eKeWbB-Fp6Ff24BXSemJOa6UEZa9IAp9U7O9oBM,1073
-gemini_agent_framework-0.1.13.dist-info/RECORD,,