Skip to content

Commit

Permalink
Merge pull request #1014 from Pythagora-io/limit-docsets
Browse files Browse the repository at this point in the history
Only add docs for non-simple epics.
  • Loading branch information
gperetin authored Jun 12, 2024
2 parents d790a6b + 1701f2e commit 9fd8ddd
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion core/agents/orchestrator.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
from core.agents.tech_writer import TechnicalWriter
from core.agents.troubleshooter import Troubleshooter
from core.db.models.project_state import TaskStatus
from core.db.models.specification import Complexity
from core.log import get_logger
from core.telemetry import telemetry
from core.ui.base import ProjectStage
Expand Down Expand Up @@ -198,7 +199,7 @@ def create_agent(self, prev_response: Optional[AgentResponse]) -> BaseAgent:
# Ask the Tech Lead to break down the initial project or feature into tasks and apply project template
return TechLead(self.state_manager, self.ui, process_manager=self.process_manager)

if state.current_task and state.docs is None:
if state.current_task and state.docs is None and state.specification.complexity != Complexity.SIMPLE:
return ExternalDocumentation(self.state_manager, self.ui)

# Current task status must be checked before Developer is called because we might want
Expand Down
2 changes: 1 addition & 1 deletion core/prompts/external-docs/select_docset.prompt
Original file line number Diff line number Diff line change
Expand Up @@ -12,4 +12,4 @@ Here is the list of available documentations:
{{ docset[0], docset[1] }}
{% endfor %}

Now, give me the list of the additional documentation that you would like to use to complete the task listed above. Return only the documentation that is absolutely required for the given task, and only from the list of available documentations provided above. If there is no additional documentation in the list that you would like to use, return an empty list.
Now, give me the list of the additional documentation that you would like to use to complete the task listed above. Return only the documentation that is absolutely required for the given task, only from the list of available documentations provided above and a MAXIMUM OF 3 ITEMS. If there is no additional documentation in the list that you would like to use, return an empty list.

0 comments on commit 9fd8ddd

Please sign in to comment.