vision-agent 0.2.67__tar.gz → 0.2.69__tar.gz

{vision_agent-0.2.67 → vision_agent-0.2.69}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.67
+Version: 0.2.69
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai
@@ -77,6 +77,9 @@ using Azure OpenAI please see the Azure setup section):
 export OPENAI_API_KEY="your-api-key"
 ```
 
+### Important Note on API Usage
+Please be aware that using the API in this project requires you to have API credits (minimum of five US dollars). This is different from the OpenAI subscription used in this chatbot. If you don't have credit, further information can be found [here](https://github.com/moutasemalakkad/vision-agent/blob/57e50b59e39fde7d943d534a8bf83c5ff6281562/README.md#L215-L216)
+
 ### Vision Agent
 #### Basic Usage
 You can interact with the agent as you would with any LLM or LMM model:
@@ -245,4 +248,14 @@ You can then run Vision Agent using the Azure OpenAI models:
 import vision_agent as va
 agent = va.agent.AzureVisionAgent()
 ```
+******************************************************************************************************************************
+#### To get started with API credits:
+
+1. Visit the[OpenAI API platform](https://beta.openai.com/signup/) to sign up for an API key.
+2. Follow the instructions to purchase and manage your API credits.
+3. Ensure your API key is correctly configured in your project settings.
+
+Failure to have sufficient API credits may result in limited or no functionality for the features that rely on the OpenAI API.
+
+For more details on managing your API usage and credits, please refer to the OpenAI API documentation.
 

{vision_agent-0.2.67 → vision_agent-0.2.69}/README.md

@@ -40,6 +40,9 @@ using Azure OpenAI please see the Azure setup section):
 export OPENAI_API_KEY="your-api-key"
 ```
 
+### Important Note on API Usage
+Please be aware that using the API in this project requires you to have API credits (minimum of five US dollars). This is different from the OpenAI subscription used in this chatbot. If you don't have credit, further information can be found [here](https://github.com/moutasemalakkad/vision-agent/blob/57e50b59e39fde7d943d534a8bf83c5ff6281562/README.md#L215-L216)
+
 ### Vision Agent
 #### Basic Usage
 You can interact with the agent as you would with any LLM or LMM model:
@@ -208,3 +211,13 @@ You can then run Vision Agent using the Azure OpenAI models:
 import vision_agent as va
 agent = va.agent.AzureVisionAgent()
 ```
+******************************************************************************************************************************
+#### To get started with API credits:
+
+1. Visit the[OpenAI API platform](https://beta.openai.com/signup/) to sign up for an API key.
+2. Follow the instructions to purchase and manage your API credits.
+3. Ensure your API key is correctly configured in your project settings.
+
+Failure to have sufficient API credits may result in limited or no functionality for the features that rely on the OpenAI API.
+
+For more details on managing your API usage and credits, please refer to the OpenAI API documentation.

{vision_agent-0.2.67 → vision_agent-0.2.69}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.67"
+version = "0.2.69"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"

{vision_agent-0.2.67 → vision_agent-0.2.69}/vision_agent/agent/vision_agent.py

@@ -324,16 +324,28 @@ def debug_code(
             "status": "started",
         }
     )
-    fixed_code_and_test = extract_json(
-        debugger(
-            FIX_BUG.format(
-                code=code,
-                tests=test,
-                result="\n".join(result.text().splitlines()[-50:]),
-                feedback=format_memory(working_memory + new_working_memory),
+
+    fixed_code_and_test = {"code": "", "test": "", "reflections": ""}
+    success = False
+    count = 0
+    while not success and count < 3:
+        try:
+            fixed_code_and_test = extract_json(
+                debugger(
+                    FIX_BUG.format(
+                        code=code,
+                        tests=test,
+                        result="\n".join(result.text().splitlines()[-50:]),
+                        feedback=format_memory(working_memory + new_working_memory),
+                    )
+                )
             )
-        )
-    )
+            success = True
+        except Exception as e:
+            _LOGGER.exception(f"Error while extracting JSON: {e}")
+
+        count += 1
+
     old_code = code
     old_test = test
 
@@ -425,12 +437,12 @@ def retrieve_tools(
         {
             "type": "tools",
             "status": "completed",
-            "payload": tool_list,
+            "payload": {v["description"]: v for v in tool_list}.values(),
         }
     )
 
     if verbosity == 2:
-        tool_desc_str = "\n".join(tool_desc)
+        tool_desc_str = "\n".join(set(tool_desc))
         _LOGGER.info(f"Tools Description:\n{tool_desc_str}")
     tool_info_set = set(tool_info)
     return "\n\n".join(tool_info_set)