Skip to content

Commit 95247fe

Browse files
committed
Update dev tool from poetry to uv.
1 parent 395e5d0 commit 95247fe

File tree

6 files changed

+92
-4488
lines changed

6 files changed

+92
-4488
lines changed

openlrc/agents.py

+2
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,8 @@ def _initialize_chatbot(self, chatbot_model: Union[str, ModelConfig], fee_limit:
6161
return chatbot_cls(model_name=chatbot_model.name, fee_limit=fee_limit, proxy=proxy, retry=4,
6262
temperature=self.TEMPERATURE, base_url_config=base_url_config,
6363
api_key=chatbot_model.api_key)
64+
else:
65+
raise ValueError(f'Invalid chatbot model type: {type(chatbot_model)}. Expected str or ModelConfig.')
6466

6567

6668
class ChunkedTranslatorAgent(Agent):

openlrc/models.py

+1-10
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,6 @@
55
from enum import Enum
66
from typing import Optional, List
77

8-
from openlrc.logger import logger
9-
108

119
class ModelProvider(Enum):
1210
ANTHROPIC = "anthropic"
@@ -423,23 +421,16 @@ def get_model(cls, model_name: str, beta: bool = False) -> ModelInfo:
423421
return model
424422

425423
# If no exact match found, try to infer provider from model name
424+
# Note the model_provider is not vital for next processing
426425
if any(name in model_name.lower() for name in ['gpt', 'openai', 'davinci', 'text-', 'curie']):
427-
provider_name = "OpenAI"
428426
default_model = cls.DefaultOpenAIModelInfo(model_name)
429427
elif any(name in model_name.lower() for name in ['claude', 'anthropic']):
430-
provider_name = "Anthropic"
431428
default_model = cls.DefaultAnthropicModelInfo(model_name)
432429
elif any(name in model_name.lower() for name in ['gemini', 'google', 'palm']):
433-
provider_name = "Google"
434430
default_model = cls.DefaultGeminiModelInfo(model_name)
435431
else:
436-
provider_name = "third-party"
437432
default_model = cls.DefaultThirdPartyModelInfo(model_name)
438433

439-
logger.warning(f"Model '{model_name}' not found in predefined models" +
440-
(" (beta)" if beta else "") +
441-
f". Using with inferred {provider_name} provider and default parameters.")
442-
443434
return default_model
444435

445436

0 commit comments

Comments
 (0)