Update retriver.py

retriver.py

@@ -24,7 +24,7 @@ embeddings_model = AzureOpenAIEmbeddings(
     openai_api_version=os.environ["AZURE_OPENAI_API_VERSION"],
 )
 
-def retriever(query):
+def retriever(query, k):
 
     namespace="gskRegIntel"
     top_k=3
@@ -41,7 +41,7 @@ def retriever(query):
         query_embedding = embeddings_model.embed_query(query)
 
         # Perform search in Pinecone
-        results = index.query(vector=query_embedding, top_k=top_k, namespace=namespace, include_metadata=True)
+        results = index.query(vector=query_embedding, top_k=k, namespace=namespace, include_metadata=True)
 
         return results.matches