xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 03:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

DocHelper

Browse Source
This commit is contained in:
mrT23
2024-09-22 15:55:18 +03:00
parent 397963257d
commit 804cb9ec1d
2 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pr_agent/settings/configuration.toml
View File

@ -183,6 +183,7 @@ final_update_message = false
[pr_help] # /help # [pr_help] # /help #
force_local_db=false force_local_db=false
num_retrieved_snippets=5
[pr_config] # /config # [pr_config] # /config #

17
pr_agent/tools/pr_help_message.py
View File

@ -36,6 +36,7 @@ class PRHelpMessage:
self.git_provider = get_git_provider_with_context(pr_url) self.git_provider = get_git_provider_with_context(pr_url)
self.ai_handler = ai_handler() self.ai_handler = ai_handler()
self.question_str = self.parse_args(args) self.question_str = self.parse_args(args)
self.num_retrieved_snippets = get_settings().get('pr_help.num_retrieved_snippets', 5)
if self.question_str: if self.question_str:
self.vars = { self.vars = {
"question": self.question_str, "question": self.question_str,
@ -91,7 +92,7 @@ class PRHelpMessage:
vectorstore = Chroma(persist_directory=temp_dir + "/chroma_db", vectorstore = Chroma(persist_directory=temp_dir + "/chroma_db",
embedding_function=embeddings) embedding_function=embeddings)
sim_results = vectorstore.similarity_search_with_score(self.question_str, k=4) sim_results = vectorstore.similarity_search_with_score(self.question_str, k=self.num_retrieved_snippets)
except Exception as e: except Exception as e:
get_logger().error(f"Error while getting sim from S3: {e}", get_logger().error(f"Error while getting sim from S3: {e}",
artifact={"traceback": traceback.format_exc()}) artifact={"traceback": traceback.format_exc()})
@ -119,7 +120,7 @@ class PRHelpMessage:
embedding_function=embeddings) embedding_function=embeddings)
# Do similarity search # Do similarity search
sim_results = vectorstore.similarity_search_with_score(self.question_str, k=4) sim_results = vectorstore.similarity_search_with_score(self.question_str, k=self.num_retrieved_snippets)
except Exception as e: except Exception as e:
get_logger().error(f"Error while getting sim from local db: {e}", get_logger().error(f"Error while getting sim from local db: {e}",
artifact={"traceback": traceback.format_exc()}) artifact={"traceback": traceback.format_exc()})
@ -137,7 +138,7 @@ class PRHelpMessage:
) )
# Do similarity search # Do similarity search
sim_results = vectorstore.similarity_search_with_score(self.question_str, k=4) sim_results = vectorstore.similarity_search_with_score(self.question_str, k=self.num_retrieved_snippets)
except Exception as e: except Exception as e:
get_logger().error(f"Error while getting sim from Pinecone db: {e}", get_logger().error(f"Error while getting sim from Pinecone db: {e}",
artifact={"traceback": traceback.format_exc()}) artifact={"traceback": traceback.format_exc()})
@ -186,6 +187,16 @@ class PRHelpMessage:
response_str = response_yaml.get('response') response_str = response_yaml.get('response')
relevant_snippets_numbers = response_yaml.get('relevant_snippets') relevant_snippets_numbers = response_yaml.get('relevant_snippets')
if not relevant_snippets_numbers:
if get_settings().config.publish_output:
answer_str = f"### Question: \n{self.question_str}\n\n"
answer_str += f"### Answer:\n\n"
answer_str += f"Could not find relevant information to answer the question. Please provide more details and try again."
self.git_provider.publish_comment(answer_str)
else:
get_logger().info(f"Could not find relevant snippets for the question: {self.question_str}")
return
# prepare the answer # prepare the answer
answer_str = "" answer_str = ""
if response_str: if response_str: