geomind-ai 1.0.2__tar.gz → 1.0.4__tar.gz

{geomind_ai-1.0.2/geomind_ai.egg-info → geomind_ai-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geomind-ai
-Version: 1.0.2
+Version: 1.0.4
 Summary: AI agent for geospatial analysis with Sentinel-2 satellite imagery
 Author: Harsh Shinde
 License-Expression: MIT

{geomind_ai-1.0.2 → geomind_ai-1.0.4}/geomind/__init__.py

@@ -3,7 +3,7 @@ GeoMind - Geospatial AI Agent
 
 """
 
-__version__ = "1.0.2"
+__version__ = "1.0.4"
 __author__ = "Harsh Shinde"
 
 from .agent import GeoMindAgent

{geomind_ai-1.0.2 → geomind_ai-1.0.4}/geomind/agent.py

@@ -12,7 +12,14 @@ from datetime import datetime
 
 from openai import OpenAI
 
-from .config import OPENROUTER_API_KEY, OPENROUTER_API_URL, OPENROUTER_MODEL
+import requests
+
+from .config import (
+    OPENROUTER_API_KEY,
+    OPENROUTER_API_URL,
+    OPENROUTER_MODEL,
+    GEOMIND_PROXY_URL,
+)
 from .tools import (
     geocode_location,
     get_bbox_from_location,
@@ -236,29 +243,31 @@ class GeoMindAgent:
     Uses OpenRouter API for access to multiple AI models.
     """
 
-    def __init__(self, model: Optional[str] = None):
+    def __init__(self, model: Optional[str] = None, api_key: Optional[str] = None):
         """
         Initialize the GeoMind agent.
 
         Args:
             model: Model name (default: xiaomi/mimo-v2-flash:free)
+            api_key: OpenRouter API key. If not provided, uses free proxy.
         """
         self.provider = "openrouter"
-        self.api_key = OPENROUTER_API_KEY
+        self.api_key = api_key or OPENROUTER_API_KEY
         self.model_name = model or OPENROUTER_MODEL
-        self.base_url = OPENROUTER_API_URL
-
-        if not self.api_key:
-            raise ValueError(
-                "OpenRouter API key required. Set OPENROUTER_API_KEY in .env file.\n"
-                "Get your API key at: https://openrouter.ai/settings/keys"
-            )
-
-        print(f"🚀 GeoMind Agent initialized with {self.model_name} (OpenRouter)")
-        print(f"   API URL: {self.base_url}")
-
-        # Create OpenAI-compatible client
-        self.client = OpenAI(base_url=self.base_url, api_key=self.api_key)
+        self.use_proxy = not self.api_key  # Use proxy if no API key provided
+
+        if self.use_proxy:
+            self.base_url = GEOMIND_PROXY_URL
+            print(f"🚀 GeoMind Agent initialized with {self.model_name} (via proxy)")
+            print(f"   Proxy URL: {self.base_url}")
+            # No OpenAI client needed for proxy
+            self.client = None
+        else:
+            self.base_url = OPENROUTER_API_URL
+            print(f"🚀 GeoMind Agent initialized with {self.model_name} (OpenRouter)")
+            print(f"   API URL: {self.base_url}")
+            # Create OpenAI-compatible client
+            self.client = OpenAI(base_url=self.base_url, api_key=self.api_key)
 
         # Chat history
         self.history = []
@@ -291,6 +300,31 @@ When users ask for imagery:
 
 Always explain what you're doing and interpret results in a helpful way."""
 
+    def _call_llm(self, messages: list, tools: list) -> dict:
+        """Call LLM via proxy or direct OpenAI client."""
+        if self.use_proxy:
+            # Use proxy endpoint
+            payload = {
+                "model": self.model_name,
+                "messages": messages,
+                "tools": tools,
+                "tool_choice": "auto",
+                "max_tokens": 4096,
+            }
+            response = requests.post(self.base_url, json=payload, timeout=120)
+            response.raise_for_status()
+            return response.json()
+        else:
+            # Use OpenAI client directly
+            response = self.client.chat.completions.create(
+                model=self.model_name,
+                messages=messages,
+                tools=tools,
+                tool_choice="auto",
+                max_tokens=4096,
+            )
+            return response.model_dump()
+
     def _execute_function(self, name: str, args: dict) -> dict:
         """Execute a function call and return the result."""
         print(f"  🔧 Executing: {name}({args})")
@@ -324,42 +358,39 @@ Always explain what you're doing and interpret results in a helpful way."""
         while iteration < max_iterations:
             iteration += 1
 
-            # Call the model
-            response = self.client.chat.completions.create(
-                model=self.model_name,
-                messages=messages,
-                tools=TOOLS,
-                tool_choice="auto",
-                max_tokens=4096,
-            )
+            # Call the model (via proxy or direct)
+            response_data = self._call_llm(messages, TOOLS)
 
-            assistant_message = response.choices[0].message
+            # Extract assistant message from response
+            choice = response_data["choices"][0]
+            assistant_message = choice["message"]
 
             # Check if there are tool calls
-            if assistant_message.tool_calls:
+            tool_calls = assistant_message.get("tool_calls", [])
+            if tool_calls:
                 # Add assistant message with tool calls to messages
                 messages.append(
                     {
                         "role": "assistant",
-                        "content": assistant_message.content or "",
+                        "content": assistant_message.get("content") or "",
                         "tool_calls": [
                             {
-                                "id": tc.id,
+                                "id": tc["id"],
                                 "type": "function",
                                 "function": {
-                                    "name": tc.function.name,
-                                    "arguments": tc.function.arguments,
+                                    "name": tc["function"]["name"],
+                                    "arguments": tc["function"]["arguments"],
                                 },
                             }
-                            for tc in assistant_message.tool_calls
+                            for tc in tool_calls
                         ],
                     }
                 )
 
                 # Execute each tool call
-                for tool_call in assistant_message.tool_calls:
-                    func_name = tool_call.function.name
-                    func_args = json.loads(tool_call.function.arguments)
+                for tool_call in tool_calls:
+                    func_name = tool_call["function"]["name"]
+                    func_args = json.loads(tool_call["function"]["arguments"])
 
                     result = self._execute_function(func_name, func_args)
 
@@ -367,13 +398,13 @@ Always explain what you're doing and interpret results in a helpful way."""
                     messages.append(
                         {
                             "role": "tool",
-                            "tool_call_id": tool_call.id,
+                            "tool_call_id": tool_call["id"],
                             "content": json.dumps(result, default=str),
                         }
                     )
             else:
                 # No tool calls, we have a final response
-                final_text = assistant_message.content or ""
+                final_text = assistant_message.get("content") or ""
 
                 # Add to history
                 self.history.append({"role": "assistant", "content": final_text})

{geomind_ai-1.0.2 → geomind_ai-1.0.4}/geomind/config.py

@@ -49,7 +49,10 @@ OUTPUT_DIR.mkdir(exist_ok=True)
 GEOCODER_USER_AGENT = "geomind_agent_v0.1"
 
 # OpenRouter API Configuration
-# OpenRouter provides access to multiple AI models via API
-OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
+# Uses Cloudflare Workers proxy to securely handle API key
+GEOMIND_PROXY_URL = "https://geomind-proxy.harshinde.workers.dev"
+
+# Users can optionally provide their own OpenRouter key to bypass proxy
+OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY", "")
 OPENROUTER_API_URL = os.getenv("OPENROUTER_API_URL", "https://openrouter.ai/api/v1")
 OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL", "xiaomi/mimo-v2-flash:free")

{geomind_ai-1.0.2 → geomind_ai-1.0.4/geomind_ai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geomind-ai
-Version: 1.0.2
+Version: 1.0.4
 Summary: AI agent for geospatial analysis with Sentinel-2 satellite imagery
 Author: Harsh Shinde
 License-Expression: MIT

{geomind_ai-1.0.2 → geomind_ai-1.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "geomind-ai"
-version = "1.0.2"
+version = "1.0.4"
 description = "AI agent for geospatial analysis with Sentinel-2 satellite imagery"
 readme = "README.md"
 authors = [