Make OpenAI conversation model configurable via khoj.yml

- Default to using `text-davinci-003' if conversation model not
  explicitly configured by user. Stop using the older `davinci' and
  `davinci-instruct' models

- Use `model' instead of `engine' as parameter.
  Usage of `engine' parameter in OpenAI API is deprecated
This commit is contained in:
Debanjum Singh Solanky
2023-01-09 00:08:03 -03:00
parent 7e05389776
commit 918af5e6f8
6 changed files with 19 additions and 12 deletions

View File

@@ -51,6 +51,7 @@ class SearchModels():
class ConversationProcessorConfigModel():
def __init__(self, processor_config: ConversationProcessorConfig):
self.openai_api_key = processor_config.openai_api_key
self.model = processor_config.model
self.conversation_logfile = Path(processor_config.conversation_logfile)
self.chat_session = ''
self.meta_log: dict = {}

View File

@@ -66,6 +66,7 @@ class SearchConfig(ConfigBase):
class ConversationProcessorConfig(ConfigBase):
openai_api_key: str
conversation_logfile: Path
model: Optional[str] = "text-davinci-003"
class ProcessorConfig(ConfigBase):
conversation: Optional[ConversationProcessorConfig]