Compare commits

..

14 Commits

View File

@ -26,6 +26,7 @@ class OpenAIConfig(AIConfig):
top_p: float top_p: float
frequency_penalty: float frequency_penalty: float
presence_penalty: float presence_penalty: float
system: str
@classmethod @classmethod
def from_dict(cls: Type[OpenAIConfigInst], source: dict[str, Any]) -> OpenAIConfigInst: def from_dict(cls: Type[OpenAIConfigInst], source: dict[str, Any]) -> OpenAIConfigInst:
@ -40,7 +41,8 @@ class OpenAIConfig(AIConfig):
temperature=float(source['temperature']), temperature=float(source['temperature']),
top_p=float(source['top_p']), top_p=float(source['top_p']),
frequency_penalty=float(source['frequency_penalty']), frequency_penalty=float(source['frequency_penalty']),
presence_penalty=float(source['presence_penalty']) presence_penalty=float(source['presence_penalty']),
system=str(source['system'])
) )
@ -61,7 +63,8 @@ class Config:
return cls( return cls(
system=str(source['system']), system=str(source['system']),
db=str(source['db']), db=str(source['db']),
openai=OpenAIConfig.from_dict(source['openai']) # FIXME: move the 'system' parameter into the OpenAI section
openai=OpenAIConfig.from_dict(source['openai'].update({'system': source['system']}))
) )
@classmethod @classmethod