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

get protection

Browse Source
This commit is contained in:
mrT23
2024-09-21 17:07:46 +03:00
parent a70fe27d94
commit 1e51acff22
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pr_agent/tools/pr_help_message.py
View File

@ -142,12 +142,12 @@ class PRHelpMessage:
if self.question_str: if self.question_str:
get_logger().info(f'Answering a PR question about the PR {self.git_provider.pr_url} ') get_logger().info(f'Answering a PR question about the PR {self.git_provider.pr_url} ')
if not get_settings().openai.key: if not get_settings().get('openai.key'):
if get_settings().config.publish_output: if get_settings().config.publish_output:
self.git_provider.publish_comment( self.git_provider.publish_comment(
"The `Help` tool chat requires an OpenAI API key, which is not configured.") "The `Help` tool chat feature requires an OpenAI API key for calculating embeddings")
else: else:
get_logger().error("The `Help` tool chat requires an OpenAI API key, which is not configured.") get_logger().error("The `Help` tool chat feature requires an OpenAI API key for calculating embeddings")
return return
# Initialize embeddings # Initialize embeddings
@ -158,7 +158,7 @@ class PRHelpMessage:
# Get similar snippets via similarity search # Get similar snippets via similarity search
if get_settings().pr_help.force_local_db: if get_settings().pr_help.force_local_db:
sim_results = self.get_sim_results_from_local_db(embeddings) sim_results = self.get_sim_results_from_local_db(embeddings)
elif get_settings().pinecone.api_key: elif get_settings().get('pinecone.api_key'):
sim_results = self.get_sim_results_from_pinecone_db(embeddings) sim_results = self.get_sim_results_from_pinecone_db(embeddings)
else: else:
sim_results = self.get_sim_results_from_s3_db(embeddings) sim_results = self.get_sim_results_from_s3_db(embeddings)