geomind-ai 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

geomind/__init__.py

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

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,
@@ -242,28 +249,25 @@ class GeoMindAgent:
 
         Args:
             model: Model name (default: xiaomi/mimo-v2-flash:free)
-            api_key: OpenRouter API key. If not provided, looks for OPENROUTER_API_KEY env variable.
+            api_key: OpenRouter API key. If not provided, uses free proxy.
         """
         self.provider = "openrouter"
         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.\n"
-                "You can provide it in three ways:\n"
-                "1. Pass it to the constructor: GeoMindAgent(api_key='your-key')\n"
-                "2. Set OPENROUTER_API_KEY environment variable\n"
-                "3. Create a .env file with OPENROUTER_API_KEY=your-key\n"
-                "\nGet 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 = []
@@ -296,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})")
@@ -329,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)
 
@@ -372,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/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.3.dist-info → geomind_ai-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geomind-ai
-Version: 1.0.3
+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.4.dist-info/RECORD

@@ -0,0 +1,14 @@
+geomind/__init__.py,sha256=l1saSR9mdB8meBsMNkp1uCJTh4y4lCsAU-iunasV6bc,151
+geomind/agent.py,sha256=KJvgEulr9zOttbTz_paZD8uDrdgDlbr9UnaqmKgdXa4,16238
+geomind/cli.py,sha256=tmMkvO9scyvzpRtiKbcb1jtWn3fSgFoywQmuusmjDuI,3283
+geomind/config.py,sha256=hgYQ33ZL9xeHzZXBFLEtx4tJX7ZHN14Ft0_ru7sHMQQ,2154
+geomind/tools/__init__.py,sha256=8iumGwIFHh8Bj1VJNgZtmKnEBqCy6_cRkzYENDUH7x4,720
+geomind/tools/geocoding.py,sha256=hiLpzHpkJP6IgWAUtZMnHL6qpkWcYWVLpGe0yfYxXv8,3007
+geomind/tools/processing.py,sha256=vMp8PMb8h8QiBRBFRvI_TGRqEDBTDQvSV0zvC1Ji5bc,9976
+geomind/tools/stac_search.py,sha256=V6230l4aHjedPWXu-3Cjmfc6diSFh5zsycewUko0W8k,6452
+geomind_ai-1.0.4.dist-info/licenses/LICENSE,sha256=aveu0ERm7I3NnIu8rtpKdvd0eyRpmktXKU0PBABtSN0,1069
+geomind_ai-1.0.4.dist-info/METADATA,sha256=Mf_JOGNdILojqcMYcxjlb-uqSnj42iISXsCCidylcv0,2233
+geomind_ai-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+geomind_ai-1.0.4.dist-info/entry_points.txt,sha256=2nPR3faYKl0-1epccvzMJ2xdi-Q1Vt7aOSvA84oIWnw,45
+geomind_ai-1.0.4.dist-info/top_level.txt,sha256=rjKWNSNRhq4R9xJoZGsG-eAaH7BmTVNvfrrbcaJMIIs,8
+geomind_ai-1.0.4.dist-info/RECORD,,

geomind_ai-1.0.3.dist-info/RECORD

@@ -1,14 +0,0 @@
-geomind/__init__.py,sha256=n5p83zzIUSJlgGvqvyilFeU5u84if0DoTwsdj4gcLNE,151
-geomind/agent.py,sha256=t5xQ5oldG5BrZfuk67RSKuDNl1TmIxwvPqbkvJqO_hE,15398
-geomind/cli.py,sha256=tmMkvO9scyvzpRtiKbcb1jtWn3fSgFoywQmuusmjDuI,3283
-geomind/config.py,sha256=hv3DNM7QbCuOSWuPmXeiVxEbGxH7fRLxOc8jexd7rrY,2011
-geomind/tools/__init__.py,sha256=8iumGwIFHh8Bj1VJNgZtmKnEBqCy6_cRkzYENDUH7x4,720
-geomind/tools/geocoding.py,sha256=hiLpzHpkJP6IgWAUtZMnHL6qpkWcYWVLpGe0yfYxXv8,3007
-geomind/tools/processing.py,sha256=vMp8PMb8h8QiBRBFRvI_TGRqEDBTDQvSV0zvC1Ji5bc,9976
-geomind/tools/stac_search.py,sha256=V6230l4aHjedPWXu-3Cjmfc6diSFh5zsycewUko0W8k,6452
-geomind_ai-1.0.3.dist-info/licenses/LICENSE,sha256=aveu0ERm7I3NnIu8rtpKdvd0eyRpmktXKU0PBABtSN0,1069
-geomind_ai-1.0.3.dist-info/METADATA,sha256=y6efOTHaisR2kWz4J4BpQLc6rLV7iMyTvKkuK2yuYM8,2233
-geomind_ai-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-geomind_ai-1.0.3.dist-info/entry_points.txt,sha256=2nPR3faYKl0-1epccvzMJ2xdi-Q1Vt7aOSvA84oIWnw,45
-geomind_ai-1.0.3.dist-info/top_level.txt,sha256=rjKWNSNRhq4R9xJoZGsG-eAaH7BmTVNvfrrbcaJMIIs,8
-geomind_ai-1.0.3.dist-info/RECORD,,