Spaces:
Sleeping
Sleeping
retrieve more file
Browse files- src/chatbot.py +2 -2
src/chatbot.py
CHANGED
|
@@ -80,12 +80,12 @@ def create_rag_chain():
|
|
| 80 |
# 2. Create retriever
|
| 81 |
retriever = vectorstore.as_retriever(
|
| 82 |
search_type="similarity",
|
| 83 |
-
search_kwargs={"k":
|
| 84 |
)
|
| 85 |
|
| 86 |
# 3. Create LLM
|
| 87 |
llm = ChatOpenAI(
|
| 88 |
-
model=os.getenv("OPEN_AI_CHAT_MODEL", "gpt-4o
|
| 89 |
temperature=0.3
|
| 90 |
)
|
| 91 |
|
|
|
|
| 80 |
# 2. Create retriever
|
| 81 |
retriever = vectorstore.as_retriever(
|
| 82 |
search_type="similarity",
|
| 83 |
+
search_kwargs={"k": 8}
|
| 84 |
)
|
| 85 |
|
| 86 |
# 3. Create LLM
|
| 87 |
llm = ChatOpenAI(
|
| 88 |
+
model=os.getenv("OPEN_AI_CHAT_MODEL", "gpt-4o"),
|
| 89 |
temperature=0.3
|
| 90 |
)
|
| 91 |
|