gemini-agent-framework 0.1.8__tar.gz → 0.1.10__tar.gz

{gemini_agent_framework-0.1.8 → gemini_agent_framework-0.1.10}/PKG-INFO

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

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

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

{gemini_agent_framework-0.1.8 → gemini_agent_framework-0.1.10}/src/gemini_agent/agent.py

@@ -143,6 +143,7 @@ class Agent:
     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.
+        If a variable with the same name exists, creates a new variable with a counter suffix.
 
         Args:
             name: The name of the variable
@@ -150,6 +151,29 @@ class Agent:
             description: A description of what the variable represents
             type_hint: Optional type hint for the variable
         """
+        # Check if the base name exists
+        if name in self._stored_variables:
+            # Find all variables that start with the base name
+            existing_vars = [var_name for var_name in self._stored_variables.keys() 
+                           if var_name.startswith(name + '_') or var_name == name]
+            
+            # Find the highest counter used
+            max_counter = 0
+            for var_name in existing_vars:
+                if var_name == name:
+                    max_counter = max(max_counter, 1)
+                else:
+                    try:
+                        counter = int(var_name.split('_')[-1])
+                        max_counter = max(max_counter, counter)
+                    except ValueError:
+                        continue
+            
+            # Create new name with incremented counter
+            new_name = f"{name}_{max_counter + 1}"
+            print(f"Variable '{name}' already exists. Creating new variable '{new_name}'")
+            name = new_name
+        
         self._stored_variables[name] = {
             'value': value,
             'description': description,
@@ -157,6 +181,8 @@ class Agent:
             'created_at': datetime.now().isoformat()
         }
 
+        return name
+
     def get_variable(self, name: str) -> Any:
         """
         Retrieves a stored variable's value.
@@ -213,7 +239,6 @@ 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.
         - When using stored variables, ALWAYS use the {{"variable": "variable_name"}} syntax
         """.format(
             tools_list="\n".join([f"- {name}: {desc}" for name, desc in 
@@ -385,15 +410,21 @@ class Agent:
                             
                             result_key = f"result_{len(self._intermediate_results)}"
                             self._intermediate_results[result_key] = function_result
-
+                            
+                            varaible_name = self.set_variable(result_key, function_result, "the result of function call with name {tool_name} and arguments {args}")
                             function_response_part = {
                                 "functionResponse": {
                                     "name": tool_name,
                                     "response": {
                                         "content": function_result,
+                                        "key": varaible_name,
+                                        "content_type": type(function_result).__name__
                                     }
                                 }
                             }
+
+                            payload["contents"].append({"role": "user", "parts": [{"text": f"the return value of the function stored in the variable {varaible_name}"}]})
+                            
                             payload["contents"].append({"role": "user", "parts": [function_response_part]})
 
                         except Exception as e:

gemini_agent_framework-0.1.8/src/gemini_agent/__init__.py

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