Skip to content

Commit 5b031b4

Browse files
authored
Use the default model for the text Chain of Thought approach (#375)
1 parent 9968168 commit 5b031b4

File tree

3 files changed

+82
-85
lines changed

3 files changed

+82
-85
lines changed

llm_core/llm_core/models/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,6 @@
1919
types.append(openai_config.OpenAIModelConfig)
2020
if default_model_name in openai_config.available_models:
2121
DefaultModelConfig = openai_config.OpenAIModelConfig
22-
if "openai_gpt-4o-mini" in openai_config.available_models:
23-
MiniModelConfig = openai_config.OpenAIModelConfig(model_name="openai_gpt-4o-mini",max_tokens=3000, temperature=0,top_p=0.9,presence_penalty=0,frequency_penalty=0)
2422
if evaluation_model_name in openai_config.available_models:
2523
evaluation_model = openai_config.available_models[evaluation_model_name]
2624
except AttributeError:
Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,14 @@
11
from pydantic import BaseModel, Field
22
from typing import Literal
3-
from llm_core.models import ModelConfigType, MiniModelConfig
3+
from llm_core.models import ModelConfigType, DefaultModelConfig
44

55
from module_text_llm.approach_config import ApproachConfig
66
from module_text_llm.chain_of_thought_approach.prompt_generate_feedback import CoTGenerateSuggestionsPrompt
77
from module_text_llm.chain_of_thought_approach.prompt_thinking import ThinkingPrompt
88

99
class ChainOfThoughtConfig(ApproachConfig):
10-
# Defaults to the cheaper mini 4o model
1110
type: Literal['chain_of_thought'] = 'chain_of_thought'
12-
model: ModelConfigType = Field(default=MiniModelConfig) # type: ignore
11+
model: ModelConfigType = Field(default=DefaultModelConfig) # type: ignore
1312
thikning_prompt: ThinkingPrompt = Field(default=ThinkingPrompt())
1413
generate_suggestions_prompt: CoTGenerateSuggestionsPrompt = Field(default=CoTGenerateSuggestionsPrompt())
1514

0 commit comments

Comments
 (0)