Update logic.py

logic.py

@@ -2,7 +2,6 @@ import kuzu
 import logging
 import sys
 import os
-from llama_index.llms import OpenAI
 #import llama_index
 from llama_index.graph_stores import KuzuGraphStore
 from llama_index import (
@@ -34,11 +33,12 @@ def save_uploadedfile(uploadedfile):
 
 def load_index(token,name):
     os.environ["OPENAI_API_KEY"] = token
+    os.environ["OPENAI_API_BASE"] = base_url
     logging.basicConfig(stream=sys.stdout, level=logging.INFO)
     
     db = kuzu.Database(name+"/kg")
     graph_store = KuzuGraphStore(db)
-    llm = OpenAI(temperature=0, model="gpt-3.5-turbo", api_key=token, openai_api_base="https://ipin-copilot.onrender.com/v1")
+    llm = OpenAI(temperature=0, model="gpt-3.5-turbo", api_key=token, openai_api_base=base_url)
 
     service_context = ServiceContext.from_defaults(llm=llm, chunk_size=512)
     storage_context = StorageContext.from_defaults(graph_store=graph_store,persist_dir=name+"/storage")
@@ -52,11 +52,12 @@ def get_index_pdf(token,name):
     print(documents)
     os.mkdir(name) 
     os.environ["OPENAI_API_KEY"] = token
+    os.environ["OPENAI_API_BASE"] = base_url
     logging.basicConfig(stream=sys.stdout, level=logging.INFO)
 
     db = kuzu.Database(name+"/kg")
     graph_store = KuzuGraphStore(db)
-    llm = OpenAI(temperature=0, model="gpt-3.5-turbo", api_key=token, openai_api_base="https://ipin-copilot.onrender.com/v1")
+    llm = OpenAI(temperature=0, model="gpt-3.5-turbo", api_key=token, openai_api_base=base_url)
     service_context = ServiceContext.from_defaults(llm=llm, chunk_size=512)
     storage_context = StorageContext.from_defaults(graph_store=graph_store)
     
@@ -74,6 +75,7 @@ def get_index_pdf(token,name):
 def get_index(links,token,name):
     os.mkdir(name) 
     os.environ["OPENAI_API_KEY"] = token
+    os.environ["OPENAI_API_BASE"] = base_url
     logging.basicConfig(stream=sys.stdout, level=logging.INFO)
     
     db = kuzu.Database(name+"/kg")
@@ -84,7 +86,7 @@ def get_index(links,token,name):
         links
     )
     
-    llm = OpenAI(temperature=0, model="gpt-3.5-turbo", api_key=token, openai_api_base="https://ipin-copilot.onrender.com/v1")
+    llm = OpenAI(temperature=0, model="gpt-3.5-turbo", api_key=token, openai_api_base=base_url)
     service_context = ServiceContext.from_defaults(llm=llm, chunk_size=512)
     storage_context = StorageContext.from_defaults(graph_store=graph_store)