chore(wren-ai-service): remove correcting (#874)

Canner · Nov 6, 2024 · 2b28dbf · 2b28dbf
1 parent c63c933
commit 2b28dbf
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/wren-ai-service/src/web/v1/routers/ask.py b/wren-ai-service/src/web/v1/routers/ask.py
@@ -53,7 +53,7 @@
    - **Path Parameter**:
      - `query_id`: The unique identifier of the query.
    - **Response**:
-     - `status`: The current status of the query (`"understanding"`, `"searching"`, `"generating"`, `"correcting"`, `"finished"`, `"failed"`, or `"stopped"`).
+     - `status`: The current status of the query (`"understanding"`, `"searching"`, `"generating"`, `"finished"`, `"failed"`, or `"stopped"`).
      - `response`: (Optional) A list of SQL results, each containing:
        - `sql`: The generated SQL statement.
        - `summary`: A summary of the SQL statement.

diff --git a/wren-ai-service/src/web/v1/services/ask.py b/wren-ai-service/src/web/v1/services/ask.py
@@ -101,7 +101,6 @@ class AskResultResponse(BaseModel):
         "understanding",
         "searching",
         "generating",
-        "correcting",
         "finished",
         "failed",
         "stopped",
@@ -273,10 +272,6 @@ async def ask(
                             "invalid_generation_results"
                         ]
                     ):
-                        self._ask_results[query_id] = AskResultResponse(
-                            status="correcting",
-                        )
-
                         sql_correction_results = await self._pipelines[
                             "sql_correction"
                         ].run(