diff --git a/src/datastructures.py b/src/datastructures.py index de54a37..13e0305 100644 --- a/src/datastructures.py +++ b/src/datastructures.py @@ -31,4 +31,8 @@ class CheckResponse(BaseModel): class OpenAiModel(Enum): - gpt51= "BR-GPT-5.1-2025-11-13-SecondOpinion-WebDemo" + gpt51 = ("GPT-5.1", "BR-GPT-5.1-2025-11-13-Misc") + + def __init__(self, public_name, deployment): + self._value_ = public_name + self.deployment = deployment diff --git a/src/llm.py b/src/llm.py index c86f512..0d4ab62 100644 --- a/src/llm.py +++ b/src/llm.py @@ -152,7 +152,7 @@ def call_openai(client, prompt: str, messages: List[dict], model: OpenAiModel = try: completion = client.chat.completions.create( - model=model.value, + model=model.deployment, messages=messages, stream=True, temperature=0.2 @@ -186,7 +186,7 @@ def call_openai_lin(client, prompt: str, messages: List[dict], model: OpenAiMode ) completion = client.chat.completions.create( - model=model.value, + model=model.deployment, messages=messages, stream=False, temperature=0.2