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

geomind/__init__.py

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

geomind/agent.py

@@ -12,7 +12,11 @@ from datetime import datetime
 
 from openai import OpenAI
 
-from .config import OPENROUTER_API_KEY, OPENROUTER_API_URL, OPENROUTER_MODEL
+from .config import (
+    OPENROUTER_API_KEY,
+    OPENROUTER_API_URL,
+    OPENROUTER_MODEL,
+)
 from .tools import (
     geocode_location,
     get_bbox_from_location,
@@ -241,8 +245,8 @@ class GeoMindAgent:
         Initialize the GeoMind agent.
 
         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.
+            model: Model name (default: google/gemini-2.0-flash-exp:free)
+            api_key: OpenRouter API key (required).
         """
         self.provider = "openrouter"
         self.api_key = api_key or OPENROUTER_API_KEY
@@ -252,15 +256,14 @@ class GeoMindAgent:
         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"
+                "Get your FREE API key at: https://openrouter.ai/settings/keys\n\n"
+                "Then provide it in one of these ways:\n"
+                "1. Run: geomind --api-key YOUR_KEY\n"
+                "2. Set environment variable: OPENROUTER_API_KEY=YOUR_KEY\n"
+                "3. Create .env file with: OPENROUTER_API_KEY=YOUR_KEY"
             )
 
-        print(f"🚀 GeoMind Agent initialized with {self.model_name} (OpenRouter)")
-        print(f"   API URL: {self.base_url}")
+        print(f"🚀 GeoMind Agent initialized with {self.model_name}")
 
         # Create OpenAI-compatible client
         self.client = OpenAI(base_url=self.base_url, api_key=self.api_key)
@@ -296,6 +299,17 @@ 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 OpenAI-compatible client."""
+        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 +343,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 +383,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/cli.py

@@ -3,12 +3,39 @@ Command-line interface for GeoMind.
 """
 
 import sys
+import os
 import argparse
+from pathlib import Path
 from typing import Optional
 
 from .agent import GeoMindAgent
 
 
+# Config file path for storing API key
+CONFIG_DIR = Path.home() / ".geomind"
+CONFIG_FILE = CONFIG_DIR / "config"
+
+
+def get_saved_api_key() -> Optional[str]:
+    """Get API key saved on user's PC."""
+    if CONFIG_FILE.exists():
+        try:
+            return CONFIG_FILE.read_text().strip()
+        except Exception:
+            return None
+    return None
+
+
+def save_api_key(api_key: str) -> bool:
+    """Save API key to user's PC."""
+    try:
+        CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+        CONFIG_FILE.write_text(api_key)
+        return True
+    except Exception:
+        return False
+
+
 def main():
     """Main CLI entry point for the geomind package."""
     parser = argparse.ArgumentParser(
@@ -56,9 +83,20 @@ Environment Variables:
     parser.add_argument(
         "--version", "-v", action="store_true", help="Show version and exit"
     )
+    parser.add_argument(
+        "--clear-key", action="store_true", help="Clear saved API key"
+    )
 
     args = parser.parse_args()
 
+    if args.clear_key:
+        if CONFIG_FILE.exists():
+            CONFIG_FILE.unlink()
+            print("✅ Saved API key cleared.")
+        else:
+            print("ℹ️ No saved API key found.")
+        sys.exit(0)
+
     if args.version:
         from . import __version__
 
@@ -68,8 +106,12 @@ Environment Variables:
     # Start interactive or single-query mode
     try:
         if args.query:
-            # Single query mode
-            agent = GeoMindAgent(model=args.model, api_key=args.api_key)
+            # Single query mode - also use saved key
+            api_key = args.api_key or get_saved_api_key()
+            if not api_key:
+                print("❌ No API key found. Run 'geomind' first to set up.")
+                sys.exit(1)
+            agent = GeoMindAgent(model=args.model, api_key=api_key)
             agent.chat(args.query)
         else:
             # Interactive mode
@@ -95,6 +137,30 @@ def run_interactive(model: Optional[str] = None, api_key: Optional[str] = None):
     print("Type 'reset' to start a new conversation")
     print("=" * 60)
 
+    # Check for API key in order: argument > env > saved file
+    from .config import OPENROUTER_API_KEY
+
+    if not api_key:
+        api_key = OPENROUTER_API_KEY
+
+    if not api_key:
+        api_key = get_saved_api_key()
+
+    if not api_key:
+        print("\n🔑 OpenRouter API key required (FREE)")
+        print("   Get yours at: https://openrouter.ai/settings/keys\n")
+        api_key = input("   Enter your API key: ").strip()
+
+        if not api_key:
+            print("\n❌ No API key provided. Exiting.")
+            return
+
+        # Save the key for future use
+        if save_api_key(api_key):
+            print("   ✅ API key saved! You won't need to enter it again.\n")
+        else:
+            print("   ⚠️ Could not save API key. You'll need to enter it next time.\n")
+
     agent = GeoMindAgent(model=model, api_key=api_key)
 
     while True:

geomind/config.py

@@ -49,7 +49,7 @@ 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")
+# Get your free API key at: https://openrouter.ai/settings/keys
+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")
+OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL", "google/gemini-2.0-flash-exp:free")

{geomind_ai-1.0.3.dist-info → geomind_ai-1.0.5.dist-info}/METADATA

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

geomind_ai-1.0.5.dist-info/RECORD

@@ -0,0 +1,14 @@
+geomind/__init__.py,sha256=xq7_AdhGGOgR1EMaBVysKaqbON5BJmSvg4msrYiO5Pg,151
+geomind/agent.py,sha256=dpHH8q-x6v0CXbnTEPlJvhLb-2AqEiU4eZT__sjLl2U,15658
+geomind/cli.py,sha256=VS5rASORRuWyXRcgUvUG2qCUX15qRG6KSvkj8EVIsbQ,5250
+geomind/config.py,sha256=MQ9eP21kljXsQRIu6q9pyc7LCrLegEfVii5jYT6X23w,2027
+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.5.dist-info/licenses/LICENSE,sha256=aveu0ERm7I3NnIu8rtpKdvd0eyRpmktXKU0PBABtSN0,1069
+geomind_ai-1.0.5.dist-info/METADATA,sha256=HriO8wTTwVxyMWY9aE9MzhQf2JrohOYFKAdmUfSDwvg,2233
+geomind_ai-1.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+geomind_ai-1.0.5.dist-info/entry_points.txt,sha256=2nPR3faYKl0-1epccvzMJ2xdi-Q1Vt7aOSvA84oIWnw,45
+geomind_ai-1.0.5.dist-info/top_level.txt,sha256=rjKWNSNRhq4R9xJoZGsG-eAaH7BmTVNvfrrbcaJMIIs,8
+geomind_ai-1.0.5.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,,