fix: changes to llm classes according to base

This commit is contained in:
Siddhant Rai
2024-04-15 19:47:24 +05:30
parent c1c69ed22b
commit 60a670ce29
6 changed files with 15 additions and 6 deletions

View File

@@ -4,10 +4,11 @@ from application.core.settings import settings
class OpenAILLM(BaseLLM):
def __init__(self, api_key):
def __init__(self, api_key, *args, **kwargs):
global openai
from openai import OpenAI
super().__init__(*args, **kwargs)
self.client = OpenAI(
api_key=api_key,
)