Skip to content
Snippets Groups Projects
Commit 707c2042 authored by Marcus Schiesser's avatar Marcus Schiesser
Browse files

fix(cl): just retrieve top-k 3 for context to prevent token exceed

parent 736b5b81
No related branches found
No related tags found
No related merge requests found
......@@ -36,7 +36,7 @@ def get_chat_engine():
# Add query tool
index = get_index()
llm = index.service_context.llm
query_engine = index.as_query_engine(similarity_top_k=5)
query_engine = index.as_query_engine(similarity_top_k=3)
query_engine_tool = QueryEngineTool.from_defaults(query_engine=query_engine)
tools.append(query_engine_tool)
......
......@@ -3,5 +3,5 @@ from app.engine.index import get_index
def get_chat_engine():
return get_index().as_chat_engine(
similarity_top_k=5, chat_mode="condense_plus_context"
similarity_top_k=3, chat_mode="condense_plus_context"
)
......@@ -29,7 +29,7 @@ async function getDataSource(llm: LLM) {
export async function createChatEngine(llm: LLM) {
const index = await getDataSource(llm);
const retriever = index.asRetriever({ similarityTopK: 5 });
const retriever = index.asRetriever({ similarityTopK: 3 });
return new ContextChatEngine({
chatModel: llm,
retriever,
......
......@@ -35,7 +35,7 @@ async function getDataSource(llm: LLM) {
export async function createChatEngine(llm: LLM) {
const index = await getDataSource(llm);
const retriever = index.asRetriever();
retriever.similarityTopK = 5;
retriever.similarityTopK = 3;
return new ContextChatEngine({
chatModel: llm,
......
......@@ -31,7 +31,7 @@ async function getDataSource(llm: LLM) {
export async function createChatEngine(llm: LLM) {
const index = await getDataSource(llm);
const retriever = index.asRetriever({ similarityTopK: 5 });
const retriever = index.asRetriever({ similarityTopK: 3 });
return new ContextChatEngine({
chatModel: llm,
retriever,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment