Ilia Tambovtsev commited on
Commit
bf58afb
·
1 Parent(s): c02655d

feat: add loading openai model

Browse files
Files changed (1) hide show
  1. src/config/model_setup.py +23 -1
src/config/model_setup.py CHANGED
@@ -15,7 +15,7 @@ class ModelConfig:
15
  def load_vsegpt(
16
  self,
17
  model: str = "vis-openai/gpt-4o-mini",
18
- temperature: float = 0.0
19
  ) -> BaseLanguageModel:
20
  """Load VSEGPT OpenAI-compatible model.
21
 
@@ -36,5 +36,27 @@ class ModelConfig:
36
  temperature=temperature
37
  )
38
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
  def load_echo_llm(self) -> EchoLLM:
40
  return EchoLLM()
 
15
  def load_vsegpt(
16
  self,
17
  model: str = "vis-openai/gpt-4o-mini",
18
+ temperature: float = 0.2
19
  ) -> BaseLanguageModel:
20
  """Load VSEGPT OpenAI-compatible model.
21
 
 
36
  temperature=temperature
37
  )
38
 
39
+ def load_openai(
40
+ self,
41
+ model: str = "gpt-4o-mini",
42
+ temperature: float = 0.2
43
+ ) -> BaseLanguageModel:
44
+ """Load OpenAI model.
45
+
46
+ Args:
47
+ model: Model identifier from vsegpt.ru/Docs/Models
48
+ temperature: Sampling temperature (0.2 = deterministic)
49
+
50
+ Returns:
51
+ Configured language model instance
52
+ """
53
+ api_key = os.environ["OPENAI_API_KEY"]
54
+
55
+ return ChatOpenAI(
56
+ model=model,
57
+ api_key=api_key,
58
+ temperature=temperature
59
+ )
60
+
61
  def load_echo_llm(self) -> EchoLLM:
62
  return EchoLLM()