Ilia Tambovtsev commited on
Commit
3705b3d
·
1 Parent(s): 8dae32e

style: remove newlined params in functions

Browse files
Files changed (1) hide show
  1. src/config/model_setup.py +7 -17
src/config/model_setup.py CHANGED
@@ -17,11 +17,10 @@ class ModelConfig:
17
  Configuration class for loading different language models.
18
  Provides methods to load various model providers.
19
  """
 
20
  def load_vsegpt(
21
- self,
22
- model: str = "vis-openai/gpt-4o-mini",
23
- temperature: float = 0.2
24
- ) -> BaseLanguageModel:
25
  """Load VSEGPT OpenAI-compatible model.
26
 
27
  Args:
@@ -35,17 +34,12 @@ class ModelConfig:
35
  api_key = os.environ["VSEGPT_API_KEY"]
36
 
37
  return ChatOpenAI(
38
- base_url=api_base,
39
- model=model,
40
- api_key=api_key,
41
- temperature=temperature
42
  )
43
 
44
  def load_openai(
45
- self,
46
- model: str = "gpt-4o-mini",
47
- temperature: float = 0.2
48
- ) -> BaseLanguageModel:
49
  """Load OpenAI model.
50
 
51
  Args:
@@ -57,11 +51,7 @@ class ModelConfig:
57
  """
58
  api_key = os.environ["OPENAI_API_KEY"]
59
 
60
- return ChatOpenAI(
61
- model=model,
62
- api_key=api_key,
63
- temperature=temperature
64
- )
65
 
66
  def load_echo_llm(self) -> EchoLLM:
67
  return EchoLLM()
 
17
  Configuration class for loading different language models.
18
  Provides methods to load various model providers.
19
  """
20
+
21
  def load_vsegpt(
22
+ self, model: str = "vis-openai/gpt-4o-mini", temperature: float = 0.2
23
+ ) -> ChatOpenAI:
 
 
24
  """Load VSEGPT OpenAI-compatible model.
25
 
26
  Args:
 
34
  api_key = os.environ["VSEGPT_API_KEY"]
35
 
36
  return ChatOpenAI(
37
+ base_url=api_base, model=model, api_key=api_key, temperature=temperature
 
 
 
38
  )
39
 
40
  def load_openai(
41
+ self, model: str = "gpt-4o-mini", temperature: float = 0.2
42
+ ) -> ChatOpenAI:
 
 
43
  """Load OpenAI model.
44
 
45
  Args:
 
51
  """
52
  api_key = os.environ["OPENAI_API_KEY"]
53
 
54
+ return ChatOpenAI(model=model, api_key=api_key, temperature=temperature)
 
 
 
 
55
 
56
  def load_echo_llm(self) -> EchoLLM:
57
  return EchoLLM()