refactor: update weaviate tool environment variable names

Co-Authored-By: root@a10k.co <root@a10k.co>

Author: devin-ai-integration[bot]

agentstack/tools/weaviate/__init__.py

@@ -22,9 +22,10 @@ def search_collection(
     Returns:
         str: JSON string containing search results
     """
-    url = os.environ.get("WEAVIATE_TOOL_URL")
-    api_key = os.environ.get("WEAVIATE_TOOL_API_KEY")
-    openai_key = os.environ.get("WEAVIATE_TOOL_OPENAI_API_KEY")
+    url = os.environ.get("WEAVIATE_URL")
+    api_key = os.environ.get("WEAVIATE_API_KEY")
+    openai_key = os.environ.get("WEAVIATE_OPENAI_API_KEY") or \
+                 os.environ.get("OPENAI_API_KEY")
 
     if not url or not api_key or not openai_key:
         raise ValueError("Missing required environment variables")
@@ -69,9 +70,10 @@ def create_collection(
     Returns:
         str: Success message
     """
-    url = os.environ.get("WEAVIATE_TOOL_URL")
-    api_key = os.environ.get("WEAVIATE_TOOL_API_KEY")
-    openai_key = os.environ.get("WEAVIATE_TOOL_OPENAI_API_KEY")
+    url = os.environ.get("WEAVIATE_URL")
+    api_key = os.environ.get("WEAVIATE_API_KEY")
+    openai_key = os.environ.get("WEAVIATE_OPENAI_API_KEY") or \
+                 os.environ.get("OPENAI_API_KEY")
 
     if not url or not api_key or not openai_key:
         raise ValueError("Missing required environment variables")

agentstack/tools/weaviate/config.json

@@ -3,9 +3,9 @@
   "url": "https://github.com/weaviate/weaviate-python-client",
   "category": "vector-store",
   "env": {
-    "WEAVIATE_TOOL_URL": null,
-    "WEAVIATE_TOOL_API_KEY": null,
-    "WEAVIATE_TOOL_OPENAI_API_KEY": null
+    "WEAVIATE_URL": null,
+    "WEAVIATE_API_KEY": null,
+    "WEAVIATE_OPENAI_API_KEY": null
   },
   "dependencies": [
     "weaviate-client>=3.0.0",