diff --git a/lars/nepho/config.py b/lars/nepho/config.py index 7d9363d..1fd1955 100644 --- a/lars/nepho/config.py +++ b/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() diff --git a/lars/nepho/models/__init__.py b/lars/nepho/models/__init__.py index 626171a..8157793 100644 --- a/lars/nepho/models/__init__.py +++ b/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"] diff --git a/lars/nepho/models/gpt_model.py b/lars/nepho/models/gpt_model.py index edf5bd5..cc87d1c 100644 --- a/lars/nepho/models/gpt_model.py +++ b/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() + diff --git a/pyproject.toml b/pyproject.toml index 1e312f1..a7ac28b 100644 --- a/pyproject.toml +++ b/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" \ No newline at end of file