ADD: Ask Sage API functionality.

lars/nepho/config.py

@@ -26,4 +26,8 @@ class Config:
     MAX_CONCURRENT_MODELS: int = int(os.getenv("MAX_CONCURRENT_MODELS", "3"))
     REQUEST_TIMEOUT: int = int(os.getenv("REQUEST_TIMEOUT", "60"))
 
+    DEFAULT_ASK_SAGE_USER_URL = "https://api.asksage.anl.gov/user"
+    DEFAULT_ASK_SAGE_SERVER_URL = "https://api.asksage.anl.gov/server"
+
+
 config = Config()

lars/nepho/models/__init__.py

@@ -1,5 +1,6 @@
 from .base_model import BaseModel
 from .gpt_model import GPTModel
 from .ollama_model import OllamaModel
+from .ask_sage_model import AskSageModel
 
-__all__ = ["BaseModel", "GPTModel", "OllamaModel"]
+__all__ = ["BaseModel", "GPTModel", "OllamaModel", "AskSageModel"]

lars/nepho/models/gpt_model.py

@@ -34,7 +34,7 @@ async def chat(self, prompt: str, images: Optional[List[str]] = None) -> str:
                     content.append({
                         "type": "image_url",
                         "image_url": {
-                            "url": f"data:image/jpeg;base64,{image_data}"
+                            "url": f"data:image/png;base64,{image_data}"
                         }
                     })
 
@@ -62,4 +62,5 @@ async def chat(self, prompt: str, images: Optional[List[str]] = None) -> str:
 
     def supports_vision(self) -> bool:
         """Check if this model supports vision capabilities."""
-        return "vision" in self.model_name.lower() or "gpt-4" in self.model_name.lower()
+        return "vision" in self.model_name.lower() or "gpt-4" in self.model_name.lower() or "gpt-5" in self.model_name.lower()
+

pyproject.toml

@@ -23,7 +23,9 @@ classifiers = [
 dependencies = []
 
 [project.optional-dependencies]
-dev = ["pytest>=6.0", "black", "flake8", "openai", "xradar", "python-dotenv", "scikit-learn", "cmweather", "torchvision", "torch", "aiohttp", "matplotlib", "pandas"]
+dev = ["pytest>=6.0", "black", "flake8", "openai", "xradar", "python-dotenv", 
+    "scikit-learn", "cmweather", "torchvision", "torch", "aiohttp", "matplotlib", "pandas",
+    "asksageclient", "pip_system_certs", "requests"]
 
 [project.urls]
 Homepage = "https://github.com/rcjackson/lars"