Merge pull request #1068 from Pythagora-io/fixes_2

Fixes 2
Pythagora-io · Jul 29, 2024 · 969e652 · 969e652
2 parents 4b4fa08 + b0e6ea7
commit 969e652
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/core/agents/task_reviewer.py b/core/agents/task_reviewer.py
@@ -47,7 +47,7 @@ async def review_code_changes(self) -> AgentResponse:
         )
         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)
diff --git a/core/templates/registry.py b/core/templates/registry.py
@@ -2,22 +2,24 @@
 
 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__)
 
 
 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,
 }
diff --git a/tests/templates/test_templates.py b/tests/templates/test_templates.py
@@ -6,6 +6,7 @@
 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):