Update functions_hf.py

functions_hf.py

@@ -1,5 +1,5 @@
 from langchain_community.utilities import SQLDatabase
-from langchain import HuggingFaceHub
+from langchain_community.llms import DeepInfra
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnablePassthrough
@@ -9,18 +9,12 @@ import os
 from dotenv import load_dotenv
 
 load_dotenv()
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
-from langchain_openai import ChatOpenAI
+api = os.getenv("DEEPINFRA_TOKEN")
+deepinfra_model = "meta-llama/Llama-3.3-70B-Instruct"
 
-llm = ChatOpenAI(
-    model="gpt-4o",
-    temperature=0,
-    max_tokens=None,
-    timeout=None,
-    max_retries=2,
-    api_key=OPENAI_API_KEY,  # if you prefer to pass api key in directly instaed of using env vars
-)
+llm = DeepInfra(model_id=deepinfra_model,
+                    deepinfra_api_token=api)
 
 
 # Initialize database connection