From ec97ebd4e74f9f302db8e71768eaa198029d58fb Mon Sep 17 00:00:00 2001
From: ali <mohammed18200118@gmail.com>
Date: Wed, 28 Jan 2026 22:23:54 +0200
Subject: [PATCH] more cleanup

---
 .idea/codeflash_internal.iml                | 49 ---------------------
 django/aiservice/languages/js_ts/testgen.py |  4 +-
 django/aiservice/testgen/models.py          |  2 +-
 django/aiservice/testgen/testgen.py         |  6 ---
 4 files changed, 3 insertions(+), 58 deletions(-)
 delete mode 100644 .idea/codeflash_internal.iml

diff --git a/.idea/codeflash_internal.iml b/.idea/codeflash_internal.iml
deleted file mode 100644
index 5081d9849..000000000
--- a/.idea/codeflash_internal.iml
+++ /dev/null
@@ -1,49 +0,0 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<module type="PYTHON_MODULE" version="4">
-  <component name="FacetManager">
-    <facet type="django" name="Django">
-      <configuration>
-        <option name="rootFolder" value="$MODULE_DIR$/django/aiservice" />
-        <option name="settingsModule" value="aiservice/settings.py" />
-        <option name="manageScript" value="manage.py" />
-        <option name="environment" value="&lt;map/&gt;" />
-        <option name="doNotUseTestRunner" value="false" />
-        <option name="trackFilePattern" value="" />
-      </configuration>
-    </facet>
-  </component>
-  <component name="NewModuleRootManager" inherit-compiler-output="true">
-    <exclude-output />
-    <content url="file://$MODULE_DIR$">
-      <sourceFolder url="file://$MODULE_DIR$/cli/tests" isTestSource="true" />
-      <sourceFolder url="file://$MODULE_DIR$/django/aiservice" isTestSource="false" />
-      <sourceFolder url="file://$MODULE_DIR$/js/cf-api" isTestSource="false" />
-      <sourceFolder url="file://$MODULE_DIR$/js/cf-webapp" isTestSource="false" />
-      <sourceFolder url="file://$MODULE_DIR$/js/common" isTestSource="false" />
-      <sourceFolder url="file://$MODULE_DIR$/cli" isTestSource="false" />
-      <sourceFolder url="file://$MODULE_DIR$/django/aiservice/tests" isTestSource="true" />
-      <excludeFolder url="file://$MODULE_DIR$/.aider.ident.cache.v1" />
-      <excludeFolder url="file://$MODULE_DIR$/.aider.tags.cache.v1" />
-      <excludeFolder url="file://$MODULE_DIR$/.mypy_cache" />
-      <excludeFolder url="file://$MODULE_DIR$/.pytest_cache" />
-      <excludeFolder url="file://$MODULE_DIR$/cli/dist" />
-      <excludeFolder url="file://$MODULE_DIR$/.aider.tags.cache.v3" />
-      <excludeFolder url="file://$MODULE_DIR$/js/cf-api/node_modules" />
-      <excludeFolder url="file://$MODULE_DIR$/js/cf-webapp/node_modules" />
-      <excludeFolder url="file://$MODULE_DIR$/js/common/node_modules" />
-      <excludeFolder url="file://$MODULE_DIR$/cli/.venv" />
-    </content>
-    <orderEntry type="jdk" jdkName="uv (codeflash-internal)" jdkType="Python SDK" />
-    <orderEntry type="sourceFolder" forTests="false" />
-    <orderEntry type="module" module-name="langchain" />
-  </component>
-  <component name="PackageRequirementsSettings">
-    <option name="requirementsPath" value="" />
-  </component>
-  <component name="TemplatesService">
-    <option name="TEMPLATE_CONFIGURATION" value="Django" />
-  </component>
-  <component name="TestRunnerService">
-    <option name="PROJECT_TEST_RUNNER" value="py.test" />
-  </component>
-</module>
\ No newline at end of file
diff --git a/django/aiservice/languages/js_ts/testgen.py b/django/aiservice/languages/js_ts/testgen.py
index 201f298de..a50406027 100644
--- a/django/aiservice/languages/js_ts/testgen.py
+++ b/django/aiservice/languages/js_ts/testgen.py
@@ -280,8 +280,8 @@ def validate_javascript_testgen_request_data(data: TestGenSchema) -> None:
         HttpError: If validation fails
 
     """
-    if data.test_framework not in ["jest", "mocha"]:
-        raise HttpError(400, "Invalid test framework for JavaScript/TypeScript. We only support jest and mocha.")
+    if data.test_framework not in ["jest"]:
+        raise HttpError(400, "Invalid test framework for JavaScript/TypeScript. We only support jest.")
     if not data.function_to_optimize:
         raise HttpError(400, "Invalid function to optimize. It is empty.")
     if not validate_trace_id(data.trace_id):
diff --git a/django/aiservice/testgen/models.py b/django/aiservice/testgen/models.py
index 273e89dd8..134b941f1 100644
--- a/django/aiservice/testgen/models.py
+++ b/django/aiservice/testgen/models.py
@@ -19,7 +19,7 @@ class TestGenSchema(Schema):
     dependent_function_names: list[str] | None = None  # Only for backwards compatibility
     module_path: str
     test_module_path: str
-    test_framework: str  # "pytest", "unittest", "jest", "mocha"
+    test_framework: str  # "pytest", "jest"
     test_timeout: int
     trace_id: str
     python_version: str | None = None  # Made optional for multi-language support
diff --git a/django/aiservice/testgen/testgen.py b/django/aiservice/testgen/testgen.py
index da20579f5..af6521017 100644
--- a/django/aiservice/testgen/testgen.py
+++ b/django/aiservice/testgen/testgen.py
@@ -465,15 +465,9 @@ async def testgen(
     request: AuthenticatedRequest, data: TestGenSchema
 ) -> tuple[int, TestGenResponseSchema | TestGenErrorResponseSchema]:
     # Route based on language
-    logging.warning(
-        f"[TESTGEN DEBUG] Received request with language='{data.language}', framework='{data.test_framework}'"
-    )
     if data.language in ("javascript", "typescript"):
-        logging.warning("[TESTGEN DEBUG] Routing to testgen_javascript")
         return await testgen_javascript(request, data)
-
     # Default: Python test generation
-    logging.warning("[TESTGEN DEBUG] Routing to testgen_python")
     return await testgen_python(request, data)