fix: tentando fix do model

langchain_backend/utils.py

@@ -6,6 +6,7 @@ from langchain_openai import OpenAIEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 from langchain.chains.combine_documents import create_stuff_documents_chain
 from langchain_core.prompts import ChatPromptTemplate
+from langchain_huggingface import HuggingFaceEndpoint
 
 os.environ.get("OPENAI_API_KEY")
 
@@ -28,9 +29,21 @@ def create_retriever(documents):
   
   return retriever
 
-def create_prompt_llm_chain(system_prompt, model="gpt-4o-mini"):
+def create_prompt_llm_chain(system_prompt, modelParam="gpt-4o-mini"):
   print('model: ', model)
-  model = ChatOpenAI(model=model)
+  if modelParam == "gpt-4o-mini":
+    model = ChatOpenAI(model=modelParam)
+  else:
+    os.environ.get("HUGGINGFACEHUB_API_TOKEN")
+
+    llm = HuggingFaceEndpoint(
+        repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
+        task="text-generation",
+        max_new_tokens=100,
+        do_sample=False,
+    )
+    result = llm.invoke("Hugging Face is")
+    print('result: ', result)
 
   system_prompt = system_prompt + "\n\n" + "{context}"
   prompt = ChatPromptTemplate.from_messages(