Jorge Londoño commited on
Commit
6e70df5
·
1 Parent(s): 426af76

Updated model

Browse files
Files changed (1) hide show
  1. correctiveRag.py +6 -6
correctiveRag.py CHANGED
@@ -24,20 +24,20 @@ logger.setLevel(logging.INFO)
24
  logger.info(f"""correctiveRag.py:Config:
25
  GROQ_MODEL = {os.getenv('GROQ_MODEL')}
26
  HUGGINGFACEHUB_EMBEDDINGS_MODEL = {os.getenv('HUGGINGFACEHUB_EMBEDDINGS_MODEL')}
27
- PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")[:5]
28
  """)
29
 
30
  # Prepare the LLM
31
- # from langchain_groq import ChatGroq
32
- # assert os.getenv('GROQ_MODEL') is not None, "GROQ_MODEL not set"
33
- # llm = ChatGroq(model_name=os.getenv('GROQ_MODEL'), temperature=0, verbose=True)
34
 
35
  # from langchain_openai import ChatOpenAI
36
  # assert os.getenv('OPENAI_MODEL_NAME') is not None, "GROQ_MODEL not set"
37
  # llm = ChatOpenAI(model=os.getenv("OPENAI_MODEL_NAME"), temperature=0.1, verbose=True)
38
 
39
- # Huggingface
40
- llm = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
41
 
42
 
43
  # Prepare the retriever
 
24
  logger.info(f"""correctiveRag.py:Config:
25
  GROQ_MODEL = {os.getenv('GROQ_MODEL')}
26
  HUGGINGFACEHUB_EMBEDDINGS_MODEL = {os.getenv('HUGGINGFACEHUB_EMBEDDINGS_MODEL')}
27
+ PINECONE_API_KEY = {os.getenv("PINECONE_API_KEY")[:5]}
28
  """)
29
 
30
  # Prepare the LLM
31
+ from langchain_groq import ChatGroq
32
+ assert os.getenv('GROQ_MODEL') is not None, "GROQ_MODEL not set"
33
+ llm = ChatGroq(model_name=os.getenv('GROQ_MODEL'), temperature=0, verbose=True)
34
 
35
  # from langchain_openai import ChatOpenAI
36
  # assert os.getenv('OPENAI_MODEL_NAME') is not None, "GROQ_MODEL not set"
37
  # llm = ChatOpenAI(model=os.getenv("OPENAI_MODEL_NAME"), temperature=0.1, verbose=True)
38
 
39
+ # Huggingface - Does not support structured_output
40
+ # llm = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
41
 
42
 
43
  # Prepare the retriever