Adding new LLM class

This commit is contained in:
João Moura
2024-09-23 03:59:05 -03:00
parent 59e51f18fd
commit a19a4a5556
9 changed files with 124 additions and 93 deletions

View File

@@ -42,6 +42,6 @@ class InternalInstructor:
if self.instructions:
messages.append({"role": "system", "content": self.instructions})
model = self._client.chat.completions.create(
model=self.llm, response_model=self.model, messages=messages
model=self.llm.model, response_model=self.model, messages=messages
)
return model