mayooear / gpt4-pdf-chatbot-langchain

GPT4 & LangChain Chatbot for large PDF docs
https://www.youtube.com/watch?v=ih9PBGVVOO4
14.95k stars 3.02k forks source link

How to include more than 4 results from Pinecone? #443

Closed troublesprouter closed 8 months ago

troublesprouter commented 12 months ago

Where can I modify the topK or number of results returned and displayed?

dosubot[bot] commented 12 months ago

🤖

Hi @troublesprouter! It's great to see you again. I hope all has been going well since our last interaction.

Based on the information provided, the Pinecone integration is implemented in the utils/pinecone-client.ts file. However, the specific function responsible for querying the Pinecone database and setting the number of results returned by a Pinecone query is not specified in the provided context.

Typically, in Pinecone, the number of results returned by a query can be controlled by a parameter called top_k in the query function. This parameter specifies the number of nearest neighbors to return.

If this is the case in the gpt4-pdf-chatbot-langchain repository, you should be able to modify the top_k parameter in the query function in the utils/pinecone-client.ts file to control the number of results returned.

However, without the specific function or the implementation details, I can't provide a more accurate answer. Could you please provide more details about the function or the implementation of the Pinecone query in the utils/pinecone-client.ts file? This will help me give you a more precise answer.

For more information about Pinecone's query function and the top_k parameter, you can refer to the Pinecone documentation.

I hope this helps! If you have any more questions, feel free to ask.

Sources


This response is meant to be useful and save you time. It is not meant to be a precise solution, but rather a starting point for your own research.

Help me be more useful! Please leave a 👍 if this is helpful and 👎 if it is irrelevant.

If you want to continue the conversation, start your reply with @dosu-bot.

troublesprouter commented 12 months ago

Found the solution:

On line 57 of chat.ts, add:

// Set the number of results to be returned
retriever.k = 10; // Replace 10 with your desired number of results

so that it now looks like this:

// Use a callback to get intermediate sources from the middle of the chain
    let resolveWithDocuments: (value: Document[]) => void;
    const documentPromise = new Promise<Document[]>((resolve) => {
      resolveWithDocuments = resolve;
    });
    const retriever = vectorStore.asRetriever({
      callbacks: [
        {
          handleRetrieverEnd(documents) {
            resolveWithDocuments(documents);
          },
        },
      ],
    });

    // Set the number of results to be returned
    retriever.k = 10; // Replace 10 with your desired number of results

    //create chain
    const chain = makeChain(retriever);
dosubot[bot] commented 9 months ago

🤖

Hi, @troublesprouter

I'm helping the gpt4-pdf-chatbot-langchain team manage their backlog and am marking this issue as stale. It seems that you have already resolved the issue by adding a line of code in the chat.ts file to set the number of results to be returned.

Could you please confirm if this issue is still relevant to the latest version of the gpt4-pdf-chatbot-langchain repository? If it is, kindly let the gpt4-pdf-chatbot-langchain team know by commenting on the issue. Otherwise, feel free to close the issue yourself, or it will be automatically closed in 7 days. Thank you!