Update model.py

ChatWithPDF · Nov 7, 2023 · 19150c7 · 19150c7
1 parent 155f6b0
commit 19150c7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/embeddings/openai/remote/model.py b/src/embeddings/openai/remote/model.py
@@ -1,14 +1,14 @@
 from request import ModelRequest
 from openai import OpenAI
-
+import os
 
 class Model:
     embedding_model = "text-embedding-ada-002"
 
     def __new__(cls, context):
         cls.context = context
         if not hasattr(cls, 'instance'):
-            cls.instance = client = OpenAI(
+            cls.client = OpenAI(
                 api_key=os.getenv("OPENAI_API_KEY"),
             )
             cls.instance = super(Model, cls).__new__(cls)
@@ -19,7 +19,7 @@ async def inference(self, request: ModelRequest):
         query = request.query
 
         if(query != None):
-            embedding = client.embeddings.create(
+            embedding = self.client.embeddings.create(
                 input=query,
                 model=self.embedding_model,
             ).data[0].embedding