Herman/Yiqun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

import from monorepo

Browse Source
This commit is contained in:
Laura Abro
2025-04-24 10:37:07 -03:00
parent 1c6fc5540b
commit 640528c746
45 changed files with 842 additions and 1613 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
worker/orca-agent/src/workflows/repoSummarizer/phases.py
View File

@ -3,9 +3,6 @@
from prometheus_swarm.workflows.base import WorkflowPhase, Workflow
class BranchCreationPhase(WorkflowPhase):
def __init__(self, workflow: Workflow, conversation_id: str = None):
super().__init__(
@ -24,24 +21,33 @@ class RepoClassificationPhase(WorkflowPhase):
prompt_name="classify_repository",
available_tools=["read_file", "list_files", "classify_repository"],
conversation_id=conversation_id,
name="Repository Classificati on",
name="Repository Classification",
)
class ReadmeGenerationPhase(WorkflowPhase):
def __init__(
self, workflow: Workflow, conversation_id: str = None, prompt_name: str = None
):
class ReadmeSectionGenerationPhase(WorkflowPhase):
def __init__(self, workflow: Workflow, conversation_id: str = None):
super().__init__(
workflow=workflow,
prompt_name=prompt_name,
prompt_name="generate_readme_section",
available_tools=[
"read_file",
"list_files",
"write_file",
"create_readme_section",
],
conversation_id=conversation_id,
name="Readme Generation",
name="Readme Section Generation",
)
class ReadmeFileCreationPhase(WorkflowPhase):
def __init__(self, workflow: Workflow, conversation_id: str = None):
super().__init__(
workflow=workflow,
prompt_name="generate_readme",
required_tool="create_readme_file",
conversation_id=conversation_id,
name="Readme File Creation",
)