diff --git a/.env.example b/.env.example index acabe04..d4bf83f 100644 --- a/.env.example +++ b/.env.example @@ -24,6 +24,9 @@ ALIBABA_API_KEY= MOONSHOT_ENDPOINT=https://api.moonshot.cn/v1 MOONSHOT_API_KEY= +UNBOUND_ENDPOINT=https://api.getunbound.ai +UNBOUND_API_KEY= + # Set to false to disable anonymized telemetry ANONYMIZED_TELEMETRY=false diff --git a/src/utils/utils.py b/src/utils/utils.py index 2590a0b..07a6730 100644 --- a/src/utils/utils.py +++ b/src/utils/utils.py @@ -23,7 +23,8 @@ PROVIDER_DISPLAY_NAMES = { "deepseek": "DeepSeek", "google": "Google", "alibaba": "Alibaba", - "moonshot": "MoonShot" + "moonshot": "MoonShot", + "unbound": "Unbound AI" } @@ -153,7 +154,6 @@ def get_llm_model(provider: str, **kwargs): base_url=base_url, api_key=api_key, ) - elif provider == "moonshot": return ChatOpenAI( model=kwargs.get("model_name", "moonshot-v1-32k-vision-preview"), @@ -161,6 +161,13 @@ def get_llm_model(provider: str, **kwargs): base_url=os.getenv("MOONSHOT_ENDPOINT"), api_key=os.getenv("MOONSHOT_API_KEY"), ) + elif provider == "unbound": + return ChatOpenAI( + model=kwargs.get("model_name", "gpt-4o-mini"), + temperature=kwargs.get("temperature", 0.0), + base_url = os.getenv("UNBOUND_ENDPOINT", "https://api.getunbound.ai"), + api_key=api_key, + ) else: raise ValueError(f"Unsupported provider: {provider}") @@ -178,6 +185,7 @@ model_names = { "mistral": ["pixtral-large-latest", "mistral-large-latest", "mistral-small-latest", "ministral-8b-latest"], "alibaba": ["qwen-plus", "qwen-max", "qwen-turbo", "qwen-long"], "moonshot": ["moonshot-v1-32k-vision-preview", "moonshot-v1-8k-vision-preview"], + "unbound": ["gemini-2.0-flash","gpt-4o-mini", "gpt-4o", "gpt-4.5-preview"] }