Merge pull request #1068 from Pythagora-io/fixes_2

Fixes 2
This commit is contained in:
LeonOstrez
2024-07-29 10:43:31 +02:00
committed by GitHub
3 changed files with 10 additions and 6 deletions

View File

@@ -47,7 +47,7 @@ class TaskReviewer(BaseAgent):
)
llm_response: str = await llm(convo, temperature=0.7)
if "done" in llm_response.strip().lower()[-7:]:
if "done" in llm_response.strip().lower()[-20:]:
return AgentResponse.done(self)
else:
return AgentResponse.task_review_feedback(self, llm_response)

View File

@@ -2,8 +2,10 @@ from enum import Enum
from core.log import get_logger
from .javascript_react import JavascriptReactProjectTemplate
from .node_express_mongoose import NodeExpressMongooseProjectTemplate
from .react_express import ReactExpressProjectTemplate
# from .react_express import ReactExpressProjectTemplate
log = get_logger(__name__)
@@ -11,13 +13,13 @@ log = get_logger(__name__)
class ProjectTemplateEnum(str, Enum):
"""Choices of available project templates."""
# JAVASCRIPT_REACT = JavascriptReactProjectTemplate.name
JAVASCRIPT_REACT = JavascriptReactProjectTemplate.name
NODE_EXPRESS_MONGOOSE = NodeExpressMongooseProjectTemplate.name
REACT_EXPRESS = ReactExpressProjectTemplate.name
# REACT_EXPRESS = ReactExpressProjectTemplate.name
PROJECT_TEMPLATES = {
# JavascriptReactProjectTemplate.name: JavascriptReactProjectTemplate,
JavascriptReactProjectTemplate.name: JavascriptReactProjectTemplate,
NodeExpressMongooseProjectTemplate.name: NodeExpressMongooseProjectTemplate,
ReactExpressProjectTemplate.name: ReactExpressProjectTemplate,
# ReactExpressProjectTemplate.name: ReactExpressProjectTemplate,
}

View File

@@ -6,6 +6,7 @@ from core.state.state_manager import StateManager
from core.templates.registry import PROJECT_TEMPLATES
@pytest.mark.skip
@pytest.mark.asyncio
@patch("core.state.state_manager.get_config")
async def test_render_react_express_sql(mock_get_config, testmanager):
@@ -30,6 +31,7 @@ async def test_render_react_express_sql(mock_get_config, testmanager):
assert "api/models/user.js" not in files
@pytest.mark.skip
@pytest.mark.asyncio
@patch("core.state.state_manager.get_config")
async def test_render_react_express_nosql(mock_get_config, testmanager):