github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-10 13:37:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
LeonOstrez
2024-09-01 22:01:50 +02:00
parent 98832324d4
commit 17f93ea7dc
2 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
core/agents/external_docs.py
View File

@@ -44,15 +44,15 @@ class ExternalDocumentation(BaseAgent):
display_name = "Documentation"
async def run(self) -> AgentResponse:
await self._store_docs([], [])
return AgentResponse.done(self)
if self.current_state.specification.example_project:
log.debug("Example project detected, no documentation selected.")
available_docsets = []
else:
available_docsets = await self._get_available_docsets()
await self._store_docs([], available_docsets)
return AgentResponse.done(self)
selected_docsets = await self._select_docsets(available_docsets)
await telemetry.trace_code_event("docsets_used", selected_docsets)

16
core/cli/main.py
View File

@@ -85,24 +85,20 @@ async def llm_api_check(ui: UIBase) -> bool:
client_class = BaseLLMClient.for_provider(llm_config.provider)
llm_client = client_class(llm_config, stream_handler=handler, error_handler=handler)
try:
await ui.send_message(
f"API check for {llm_config.provider.value} {llm_config.model} !",
source=pythagora_source,
)
resp = await llm_client.api_check()
if not resp:
log.warning(f"API check for {llm_config.provider.value} failed.")
return False
else:
await ui.send_message(
f"DONE {llm_config.provider.value} {llm_config.model} !",
f"API check for {llm_config.provider.value} {llm_config.model} failed.",
source=pythagora_source,
)
log.info(f"API check for {llm_config.provider.value} succeeded.")
log.warning(f"API check for {llm_config.provider.value} {llm_config.model} failed.")
return False
else:
log.info(f"API check for {llm_config.provider.value} {llm_config.model} succeeded.")
return True
except APIError as err:
await ui.send_message(
f"API check for {llm_config.provider.value} failed with: {err}",
f"API check for {llm_config.provider.value} {llm_config.model} failed with: {err}",
source=pythagora_source,
)
log.warning(f"API check for {llm_config.provider.value} failed with: {err}")