Upload app.py with huggingface_hub

app.py

@@ -113,15 +113,15 @@ full_prompt = ChatPromptTemplate.from_messages(
 # )
 
 
-# llm = ChatOpenAI(
-#     model="gpt-4o-mini",
-#     temperature=0,
-#     max_tokens=None,
-#     timeout=None,
-#     max_retries=5,
-#     api_key="gl-U2FsdGVkX1/DTLQlsmj+RdJjPy3igB9qINuaX940XtJ0CPnGc/5sbBkPKah/C829",  # if you prefer to pass api key in directly instaed of using env vars
-#     base_url="https://aibe.mygreatlearning.com/openai/v1"
-# )
+llm = ChatOpenAI(
+    model="gpt-4o-mini",
+    temperature=0,
+    max_tokens=None,
+    timeout=None,
+    max_retries=5,
+    api_key="gl-U2FsdGVkX1/DTLQlsmj+RdJjPy3igB9qINuaX940XtJ0CPnGc/5sbBkPKah/C829",  # if you prefer to pass api key in directly instaed of using env vars
+    base_url="https://aibe.mygreatlearning.com/openai/v1"
+)
 
 client = OpenAI(
     api_key="gl-U2FsdGVkX1/DTLQlsmj+RdJjPy3igB9qINuaX940XtJ0CPnGc/5sbBkPKah/C829",
@@ -131,7 +131,7 @@ client = OpenAI(
 
 # Create the SQL agent using the ChatOpenAI model, database, and prompt template
 sqlite_agent = create_sql_agent(
-    llm=client,
+    llm=llm,
     db=db,
     prompt=full_prompt,
     agent_type="openai-tools",
@@ -217,10 +217,10 @@ retriever = vector_store.as_retriever(
     search_kwargs={'k': 5}
 )
 
-# client = OpenAI(
-#     api_key=api_key,
-#     base_url=endpoint
-# )
+client = OpenAI(
+    api_key=api_key,
+    base_url=endpoint
+)
 
 @tool
 def rag(user_input: str) -> str: