geomind-ai 1.0.0__tar.gz → 1.0.1__tar.gz

{geomind_ai-1.0.0/geomind_ai.egg-info → geomind_ai-1.0.1}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: geomind-ai
-Version: 1.0.0
-Summary: AI agent for geospatial analysis
+Version: 1.0.1
+Summary: AI agent for geospatial analysis with Sentinel-2 satellite imagery
 Author: Harsh Shinde
 License-Expression: MIT
-Project-URL: Homepage, https://harshshinde0.github.io/GeoMind/
+Project-URL: Homepage, https://github.com/HarshShinde0/GeoMind
 Project-URL: Repository, https://github.com/HarshShinde0/GeoMind
 Project-URL: Documentation, https://github.com/HarshShinde0/GeoMind#readme
 Project-URL: Issues, https://github.com/HarshShinde0/GeoMind/issues
@@ -36,27 +36,20 @@ Requires-Dist: numpy>=1.26.0
 Requires-Dist: python-dotenv>=1.0.0
 Dynamic: license-file
 
-### 1. Install Dependencies
+###  Install Dependencies
 
 ```bash
 pip install -r requirements.txt
 ```
 
-### 2. Set Up API Key
-
-Set your HuggingFace API key in the environment or update `config.py`:
-
-```python
-# In geomind/config.py
-HF_API_KEY = "your_huggingface_api_key"
-```
-
-Get a free API key from [HuggingFace](https://huggingface.co/settings/tokens).
-
-### 3. Run the Agent
+###  Run the Agent
 
 ```bash
-python main.py
+# Interactive mode
+geomind
+
+# Single query
+geomind --query "Find recent imagery of Paris"
 ```
 
 ## Example Queries

{geomind_ai-1.0.0 → geomind_ai-1.0.1}/README.md

@@ -1,24 +1,17 @@
-### 1. Install Dependencies
+###  Install Dependencies
 
 ```bash
 pip install -r requirements.txt
 ```
 
-### 2. Set Up API Key
-
-Set your HuggingFace API key in the environment or update `config.py`:
-
-```python
-# In geomind/config.py
-HF_API_KEY = "your_huggingface_api_key"
-```
-
-Get a free API key from [HuggingFace](https://huggingface.co/settings/tokens).
-
-### 3. Run the Agent
+###  Run the Agent
 
 ```bash
-python main.py
+# Interactive mode
+geomind
+
+# Single query
+geomind --query "Find recent imagery of Paris"
 ```
 
 ## Example Queries

geomind_ai-1.0.1/geomind/__init__.py

@@ -0,0 +1,11 @@
+"""
+GeoMind - Geospatial AI Agent
+
+"""
+
+__version__ = "1.0.1"
+__author__ = "Harsh Shinde, Rajat Shinde"
+
+from .agent import GeoMindAgent
+
+__all__ = ["GeoMindAgent"]

{geomind_ai-1.0.0 → geomind_ai-1.0.1}/geomind/agent.py

@@ -6,12 +6,15 @@ queries about satellite imagery and execute the appropriate tools.
 """
 
 import json
-from typing import Optional
+import re
+from typing import Optional, Callable, Any
 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,
@@ -48,12 +51,12 @@ TOOLS = [
                 "properties": {
                     "place_name": {
                         "type": "string",
-                        "description": "The name of the place to geocode (e.g., 'New York City', 'Paris, France')",
+                        "description": "The name of the place to geocode (e.g., 'New York City', 'Paris, France')"
                     }
                 },
-                "required": ["place_name"],
-            },
-        },
+                "required": ["place_name"]
+            }
+        }
     },
     {
         "type": "function",
@@ -65,16 +68,16 @@ TOOLS = [
                 "properties": {
                     "place_name": {
                         "type": "string",
-                        "description": "The name of the place",
+                        "description": "The name of the place"
                     },
                     "buffer_km": {
                         "type": "number",
-                        "description": "Buffer distance in kilometers (default: 10)",
-                    },
+                        "description": "Buffer distance in kilometers (default: 10)"
+                    }
                 },
-                "required": ["place_name"],
-            },
-        },
+                "required": ["place_name"]
+            }
+        }
     },
     {
         "type": "function",
@@ -87,28 +90,28 @@ TOOLS = [
                     "bbox": {
                         "type": "array",
                         "items": {"type": "number"},
-                        "description": "Bounding box as [min_lon, min_lat, max_lon, max_lat]",
+                        "description": "Bounding box as [min_lon, min_lat, max_lon, max_lat]"
                     },
                     "start_date": {
                         "type": "string",
-                        "description": "Start date in YYYY-MM-DD format",
+                        "description": "Start date in YYYY-MM-DD format"
                     },
                     "end_date": {
                         "type": "string",
-                        "description": "End date in YYYY-MM-DD format",
+                        "description": "End date in YYYY-MM-DD format"
                     },
                     "max_cloud_cover": {
                         "type": "number",
-                        "description": "Maximum cloud cover percentage (0-100)",
+                        "description": "Maximum cloud cover percentage (0-100)"
                     },
                     "max_items": {
                         "type": "integer",
-                        "description": "Maximum number of results",
-                    },
+                        "description": "Maximum number of results"
+                    }
                 },
-                "required": [],
-            },
-        },
+                "required": []
+            }
+        }
     },
     {
         "type": "function",
@@ -120,24 +123,24 @@ TOOLS = [
                 "properties": {
                     "location_name": {
                         "type": "string",
-                        "description": "Name of the location to search",
+                        "description": "Name of the location to search"
                     },
                     "days": {
                         "type": "integer",
-                        "description": "Number of days to look back (default: 7)",
+                        "description": "Number of days to look back (default: 7)"
                     },
                     "max_cloud_cover": {
                         "type": "number",
-                        "description": "Maximum cloud cover percentage",
+                        "description": "Maximum cloud cover percentage"
                     },
                     "max_items": {
                         "type": "integer",
-                        "description": "Maximum number of results",
-                    },
+                        "description": "Maximum number of results"
+                    }
                 },
-                "required": [],
-            },
-        },
+                "required": []
+            }
+        }
     },
     {
         "type": "function",
@@ -147,11 +150,14 @@ TOOLS = [
             "parameters": {
                 "type": "object",
                 "properties": {
-                    "item_id": {"type": "string", "description": "The STAC item ID"}
+                    "item_id": {
+                        "type": "string",
+                        "description": "The STAC item ID"
+                    }
                 },
-                "required": ["item_id"],
-            },
-        },
+                "required": ["item_id"]
+            }
+        }
     },
     {
         "type": "function",
@@ -163,20 +169,20 @@ TOOLS = [
                 "properties": {
                     "zarr_url": {
                         "type": "string",
-                        "description": "URL to the SR_10m Zarr asset from a STAC item",
+                        "description": "URL to the SR_10m Zarr asset from a STAC item"
                     },
                     "output_path": {
                         "type": "string",
-                        "description": "Optional path to save the output image",
+                        "description": "Optional path to save the output image"
                     },
                     "subset_size": {
                         "type": "integer",
-                        "description": "Size to subset the image (default: 1000 pixels)",
-                    },
+                        "description": "Size to subset the image (default: 1000 pixels)"
+                    }
                 },
-                "required": ["zarr_url"],
-            },
-        },
+                "required": ["zarr_url"]
+            }
+        }
     },
     {
         "type": "function",
@@ -188,20 +194,20 @@ TOOLS = [
                 "properties": {
                     "zarr_url": {
                         "type": "string",
-                        "description": "URL to the SR_10m Zarr asset",
+                        "description": "URL to the SR_10m Zarr asset"
                     },
                     "output_path": {
                         "type": "string",
-                        "description": "Optional path to save the NDVI image",
+                        "description": "Optional path to save the NDVI image"
                     },
                     "subset_size": {
                         "type": "integer",
-                        "description": "Size to subset the image",
-                    },
+                        "description": "Size to subset the image"
+                    }
                 },
-                "required": ["zarr_url"],
-            },
-        },
+                "required": ["zarr_url"]
+            }
+        }
     },
     {
         "type": "function",
@@ -213,67 +219,65 @@ TOOLS = [
                 "properties": {
                     "zarr_url": {
                         "type": "string",
-                        "description": "URL to the Zarr asset",
+                        "description": "URL to the Zarr asset"
                     },
                     "bands": {
                         "type": "array",
                         "items": {"type": "string"},
-                        "description": "List of band names to analyze",
-                    },
+                        "description": "List of band names to analyze"
+                    }
                 },
-                "required": ["zarr_url"],
-            },
-        },
-    },
+                "required": ["zarr_url"]
+            }
+        }
+    }
 ]
 
 
 class GeoMindAgent:
     """
     GeoMind - An AI agent for geospatial analysis with Sentinel-2 imagery.
-
+    
     Uses OpenRouter API for access to multiple AI models.
     """
-
-    def __init__(self, model: Optional[str] = None, api_key: Optional[str] = None):
+    
+    def __init__(self, model: 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, looks for OPENROUTER_API_KEY env variable.
         """
         self.provider = "openrouter"
-        self.api_key = api_key or OPENROUTER_API_KEY
+        self.api_key = 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"
+                "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.client = OpenAI(
+            base_url=self.base_url,
+            api_key=self.api_key
+        )
+        
         # Chat history
         self.history = []
-
+        
         # Add system message
         self.system_prompt = self._get_system_prompt()
-
+    
     def _get_system_prompt(self) -> str:
         """Get the system prompt for the agent."""
-        return f"""You are GeoMind, an expert AI assistant specialized in geospatial analysis
-and satellite imagery. You help users find, analyze, and visualize Sentinel-2 satellite data
+        return f"""You are GeoMind, an expert AI assistant specialized in geospatial analysis 
+and satellite imagery. You help users find, analyze, and visualize Sentinel-2 satellite data 
 from the EOPF (ESA Earth Observation Processing Framework) catalog.
 
 Your capabilities include:
@@ -294,20 +298,20 @@ When users ask for imagery:
 3. Offer to create visualizations if data is found
 
 Always explain what you're doing and interpret results in a helpful way."""
-
+    
     def _execute_function(self, name: str, args: dict) -> dict:
         """Execute a function call and return the result."""
         print(f"  🔧 Executing: {name}({args})")
-
+        
         if name not in TOOL_FUNCTIONS:
             return {"error": f"Unknown function: {name}"}
-
+        
         try:
             result = TOOL_FUNCTIONS[name](**args)
             return result
         except Exception as e:
             return {"error": str(e)}
-
+    
     def chat(self, message: str, verbose: bool = True) -> str:
         """
         Send a message to the agent and get a response.
@@ -315,19 +319,19 @@ Always explain what you're doing and interpret results in a helpful way."""
         if verbose:
             print(f"\n💬 User: {message}")
             print("🤔 Processing...")
-
+        
         # Add user message to history
         self.history.append({"role": "user", "content": message})
-
+        
         # Build messages with system prompt
         messages = [{"role": "system", "content": self.system_prompt}] + self.history
-
+        
         max_iterations = 10
         iteration = 0
-
+        
         while iteration < max_iterations:
             iteration += 1
-
+            
             # Call the model
             response = self.client.chat.completions.create(
                 model=self.model_name,
@@ -336,59 +340,55 @@ Always explain what you're doing and interpret results in a helpful way."""
                 tool_choice="auto",
                 max_tokens=4096,
             )
-
+            
             assistant_message = response.choices[0].message
-
+            
             # Check if there are tool calls
             if assistant_message.tool_calls:
                 # Add assistant message with tool calls to messages
-                messages.append(
-                    {
-                        "role": "assistant",
-                        "content": assistant_message.content or "",
-                        "tool_calls": [
-                            {
-                                "id": tc.id,
-                                "type": "function",
-                                "function": {
-                                    "name": tc.function.name,
-                                    "arguments": tc.function.arguments,
-                                },
+                messages.append({
+                    "role": "assistant",
+                    "content": assistant_message.content or "",
+                    "tool_calls": [
+                        {
+                            "id": tc.id,
+                            "type": "function",
+                            "function": {
+                                "name": tc.function.name,
+                                "arguments": tc.function.arguments
                             }
-                            for tc in assistant_message.tool_calls
-                        ],
-                    }
-                )
-
+                        }
+                        for tc in assistant_message.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)
-
+                    
                     result = self._execute_function(func_name, func_args)
-
+                    
                     # Add tool result to messages
-                    messages.append(
-                        {
-                            "role": "tool",
-                            "tool_call_id": tool_call.id,
-                            "content": json.dumps(result, default=str),
-                        }
-                    )
+                    messages.append({
+                        "role": "tool",
+                        "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 ""
-
+                
                 # Add to history
                 self.history.append({"role": "assistant", "content": final_text})
-
+                
                 if verbose:
                     print(f"\n🌍 GeoMind: {final_text}")
-
+                
                 return final_text
-
+        
         return "Max iterations reached."
-
+    
     def reset(self):
         """Reset the chat session."""
         self.history = []
@@ -398,7 +398,7 @@ Always explain what you're doing and interpret results in a helpful way."""
 def main(model: Optional[str] = None):
     """Main entry point for CLI usage."""
     import sys
-
+    
     print("=" * 60)
     print("🌍 GeoMind - Geospatial AI Agent")
     print("=" * 60)
@@ -406,7 +406,7 @@ def main(model: Optional[str] = None):
     print("Type 'quit' or 'exit' to end the session")
     print("Type 'reset' to start a new conversation")
     print("=" * 60)
-
+    
     try:
         agent = GeoMindAgent(model=model)
     except ValueError as e:
@@ -416,24 +416,24 @@ def main(model: Optional[str] = None):
         print(f"\n❌ Error: {e}")
         print("\nPlease check your API key and internet connection.")
         sys.exit(1)
-
+    
     while True:
         try:
             user_input = input("\n💬 You: ").strip()
-
+            
             if not user_input:
                 continue
-
-            if user_input.lower() in ["quit", "exit", "q"]:
+            
+            if user_input.lower() in ['quit', 'exit', 'q']:
                 print("\n👋 Goodbye!")
                 break
-
-            if user_input.lower() == "reset":
+            
+            if user_input.lower() == 'reset':
                 agent.reset()
                 continue
-
+            
             agent.chat(user_input)
-
+            
         except KeyboardInterrupt:
             print("\n\n👋 Goodbye!")
             break

{geomind_ai-1.0.0 → geomind_ai-1.0.1}/geomind/cli.py

@@ -32,36 +32,34 @@ Environment Variables:
   OPENROUTER_API_KEY    Your OpenRouter API key
   OPENROUTER_MODEL      Model to use (default: xiaomi/mimo-v2-flash:free)
   OPENROUTER_API_URL    API endpoint (default: https://openrouter.ai/api/v1)
-        """,
+        """
     )
 
     parser.add_argument(
-        "--query",
-        "-q",
+        "--query", "-q",
         type=str,
-        help="Single query to run (if not provided, starts interactive mode)",
+        help="Single query to run (if not provided, starts interactive mode)"
     )
     parser.add_argument(
-        "--model",
-        "-m",
+        "--model", "-m",
         type=str,
-        help="Model name to use (e.g., 'anthropic/claude-3.5-sonnet')",
+        help="Model name to use (e.g., 'anthropic/claude-3.5-sonnet')"
     )
     parser.add_argument(
-        "--api-key",
-        "-k",
+        "--api-key", "-k",
         type=str,
-        help="OpenRouter API key (or set OPENROUTER_API_KEY env variable)",
+        help="OpenRouter API key (or set OPENROUTER_API_KEY env variable)"
     )
     parser.add_argument(
-        "--version", "-v", action="store_true", help="Show version and exit"
+        "--version", "-v",
+        action="store_true",
+        help="Show version and exit"
     )
 
     args = parser.parse_args()
 
     if args.version:
         from . import __version__
-
         print(f"GeoMind version {__version__}")
         sys.exit(0)
 
@@ -104,11 +102,11 @@ def run_interactive(model: Optional[str] = None, api_key: Optional[str] = None):
             if not user_input:
                 continue
 
-            if user_input.lower() in ["quit", "exit", "q"]:
+            if user_input.lower() in ['quit', 'exit', 'q']:
                 print("\n👋 Goodbye!")
                 break
 
-            if user_input.lower() == "reset":
+            if user_input.lower() == 'reset':
                 agent.reset()
                 continue