gemini-agent-framework 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

gemini_agent/__init__.py

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

gemini_agent/agent.py

@@ -4,6 +4,7 @@ import inspect
 from functools import wraps
 from typing import List, Callable, Dict, Any, Optional
 from dotenv import load_dotenv
+from datetime import datetime
 
 load_dotenv()
 
@@ -76,6 +77,7 @@ class Agent:
         self._tool_functions: Dict[str, Callable] = {}  # Map name to actual function
         self._tool_instances: Dict[str, Any] = {}  # Store instances for class methods
         self._intermediate_results: Dict[str, Any] = {}  # Store intermediate results
+        self._stored_variables: Dict[str, Dict[str, Any]] = {}  # Store variables with metadata
 
         if tools:
             self._process_tools(tools)
@@ -138,8 +140,52 @@ class Agent:
 
             self._registered_tools_json.append(declaration_json)
 
+    def set_variable(self, name: str, value: Any, description: str = "", type_hint: type = None) -> None:
+        """
+        Stores a variable in the agent's memory with metadata.
+
+        Args:
+            name: The name of the variable
+            value: The actual value to store
+            description: A description of what the variable represents
+            type_hint: Optional type hint for the variable
+        """
+        self._stored_variables[name] = {
+            'value': value,
+            'description': description,
+            'type': type_hint or type(value).__name__,
+            'created_at': datetime.now().isoformat()
+        }
+
+    def get_variable(self, name: str) -> Any:
+        """
+        Retrieves a stored variable's value.
+
+        Args:
+            name: The name of the variable to retrieve
+
+        Returns:
+            The stored value or None if not found
+        """
+        return self._stored_variables.get(name, {}).get('value')
+
+    def list_variables(self) -> Dict[str, Dict[str, Any]]:
+        """
+        Returns information about all stored variables.
+
+        Returns:
+            Dictionary of variable names to their metadata
+        """
+        return {name: {k: v for k, v in data.items() if k != 'value'} 
+                for name, data in self._stored_variables.items()}
+
     def _get_system_prompt(self) -> str:
         """Returns a system prompt that guides the model in breaking down complex operations."""
+        variables_info = "\n".join([
+            f"- {name}: {data['description']} (Type: {data['type']})"
+            for name, data in self._stored_variables.items()
+        ])
+        
         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
@@ -152,13 +198,50 @@ class Agent:
         Available tools:
         {tools_list}
         
+        Available variables:
+        {variables_list}
+        
+        IMPORTANT - Variable Usage:
+        When you need to use a stored variable in a function call, you MUST use the following syntax:
+        - For function arguments: {{"variable": "variable_name"}}
+        - For example, if you want to use the 'current_user' variable in a function call:
+          {{"user_id": {{"variable": "current_user"}}}}
+        
         Remember:
         - Always perform one operation at a time
         - Use intermediate results from previous steps
         - If a step requires multiple tools, execute them sequentially
         - If you're unsure about the next step, explain your reasoning
-        """.format(tools_list="\n".join([f"- {name}: {desc}" for name, desc in 
-            [(tool['name'], tool['description']) for tool in self._registered_tools_json]]))
+        - You can use both stored variables and values from the prompt
+        - You don't have the ability to set new varaibles during the conversation
+        - When using stored variables, ALWAYS use the {{"variable": "variable_name"}} syntax
+        """.format(
+            tools_list="\n".join([f"- {name}: {desc}" for name, desc in 
+                [(tool['name'], tool['description']) for tool in self._registered_tools_json]]),
+            variables_list=variables_info
+        )
+
+    def _substitute_variables(self, args: Dict[str, Any]) -> Dict[str, Any]:
+        """
+        Substitutes stored variable values in function arguments.
+
+        Args:
+            args: Dictionary of argument names to values
+
+        Returns:
+            Dictionary with variable values substituted where applicable
+        """
+        substituted_args = {}
+        for name, value in args.items():
+            if isinstance(value, dict) and "variable" in value:
+                var_name = value["variable"]
+                if var_name in self._stored_variables:
+                    substituted_args[name] = self._stored_variables[var_name]['value']
+                else:
+                    substituted_args[name] = value
+            else:
+                substituted_args[name] = value
+        return substituted_args
 
     def _call_gemini_api(self, payload: Dict[str, Any]) -> Dict[str, Any]:
         """Makes a POST request to the Gemini generateContent endpoint."""
@@ -210,7 +293,7 @@ class Agent:
         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."}]})
+            current_contents.append({'role': 'model', 'parts': [{'text': "I understand I should break down complex tasks into sequential steps using the available tools and variables."}]})
 
         current_contents.append({'role': 'user', 'parts': [{'text': user_prompt}]})
         payload: Dict[str, Any] = {"contents": current_contents}
@@ -241,8 +324,8 @@ class Agent:
             final_response_schema = response_structure
 
         while True:
-            response_data = self._call_gemini_api(payload)
 
+            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')}")
                 return response_data
@@ -285,6 +368,8 @@ class Agent:
                             tool_function = self._tool_functions[tool_name]
                             print(f"--- Calling Function: {tool_name}({args}) ---")
                             
+                            # Substitute both stored variables and intermediate results
+                            args = self._substitute_variables(args)
                             for key, value in args.items():
                                 if isinstance(value, str) and value.startswith('$'):
                                     result_key = value[1:]
@@ -385,8 +470,8 @@ class Agent:
                                     print(f"Warning: Failed to parse structured output after formatting call: {e}. Returning intermediate text.")
                                     return final_text
                             return final_text
-
                 continue
+                
 
             except (KeyError, IndexError) as e:
                 print(f"Error parsing API response structure: {e}. Response: {response_data}")

{gemini_agent_framework-0.1.2.dist-info → gemini_agent_framework-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-agent-framework
-Version: 0.1.2
+Version: 0.1.4
 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.dist-info/RECORD

@@ -0,0 +1,5 @@
+gemini_agent/__init__.py,sha256=3nn6YdImhRa8SN7VOAJVo2lPbBu2cYjzxgInerYhDvc,71
+gemini_agent/agent.py,sha256=tMsgexMEsKVvfVTlluihJay4Dadkk6LWf84qe-rv4qc,24146
+gemini_agent_framework-0.1.4.dist-info/METADATA,sha256=V2RVV60ovE07FCsSfXA3mRKcr5YV6ajU-qvaQ4jeLFQ,2386
+gemini_agent_framework-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+gemini_agent_framework-0.1.4.dist-info/RECORD,,

gemini_agent_framework-0.1.2.dist-info/RECORD

@@ -1,5 +0,0 @@
-gemini_agent/__init__.py,sha256=-F0v5szfo7HJrUloNdX7J13qO1d3U2wAfxrwt1bJpUQ,71
-gemini_agent/agent.py,sha256=0slDSnf__jp8j7jxeow12O1XszdTweX2MDorr-JlAmc,20674
-gemini_agent_framework-0.1.2.dist-info/METADATA,sha256=savJnjAKvUPmN2rMRSohQzdxla1sIPfbmuEJ-YlrHU0,2386
-gemini_agent_framework-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-gemini_agent_framework-0.1.2.dist-info/RECORD,,