Spaces:
Sleeping
Sleeping
Jorge Londoño
commited on
Commit
·
6e70df5
1
Parent(s):
426af76
Updated model
Browse files- 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 |
-
|
| 32 |
-
|
| 33 |
-
|
| 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
|