replace non-openai-compatible models with openai

This commit is contained in:
Rajat Ahuja 2025-05-27 13:12:06 -04:00
parent c22cc507b0
commit 809db5c570
4 changed files with 7 additions and 7 deletions

View File

@ -26,8 +26,8 @@ logger = logging.getLogger(__name__)
USER_REPRESENTATION_METAMESSAGE_TYPE = "honcho_user_representation"
DEF_DIALECTIC_PROVIDER = ModelProvider.ANTHROPIC
DEF_DIALECTIC_MODEL = "claude-3-7-sonnet-20250219"
DEF_DIALECTIC_PROVIDER = ModelProvider.OPENAI # patching to openai for mocking with muk
DEF_DIALECTIC_MODEL = "gpt-4o"
DEF_QUERY_GENERATION_PROVIDER = ModelProvider.GROQ
DEF_QUERY_GENERATION_MODEL = "llama-3.1-8b-instant"

View File

@ -13,8 +13,8 @@ from src.utils.model_client import ModelClient, ModelProvider
logger = logging.getLogger(__name__)
# Constants for fact extraction
FACT_EXTRACTION_PROVIDER = ModelProvider.GEMINI
FACT_EXTRACTION_MODEL = "gemini-2.0-flash-lite"
FACT_EXTRACTION_PROVIDER = ModelProvider.OPENAI # patching to openai for mocking with muk
FACT_EXTRACTION_MODEL = "gpt-4o"
USER_REPRESENTATION_PROVIDER = ModelProvider.GROQ
USER_REPRESENTATION_MODEL = "llama-3.3-70b-versatile"

View File

@ -38,8 +38,8 @@ class SummaryType(Enum):
# Default model settings for summary generation
DEFAULT_PROVIDER = ModelProvider.GEMINI
DEFAULT_MODEL = "gemini-2.0-flash-lite"
DEFAULT_PROVIDER = ModelProvider.OPENAI # patching to openai for mocking with muk
DEFAULT_MODEL = "gpt-4o"
async def get_session_summaries(

View File

@ -64,7 +64,7 @@ class ModelClient:
def __init__(
self,
provider: ModelProvider = ModelProvider.ANTHROPIC,
provider: ModelProvider = ModelProvider.OPENAI, # patching default provider to openai for mocking with muk
model: Optional[str] = None,
api_key: Optional[str] = None,
base_url: Optional[str] = None,