add user_id

Canner · Sep 19, 2024 · 5f5c5d8 · 5f5c5d8
1 parent 04b88a4
commit 5f5c5d8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/wren-ai-service/src/web/v1/services/sql_answer.py b/wren-ai-service/src/web/v1/services/sql_answer.py
@@ -18,6 +18,7 @@ class SqlAnswerRequest(BaseModel):
     sql: str
     sql_summary: str
     thread_id: Optional[str] = None
+    user_id: Optional[str] = None
 
     @property
     def query_id(self) -> str:

diff --git a/wren-ai-service/src/web/v1/services/sql_explanation.py b/wren-ai-service/src/web/v1/services/sql_explanation.py
@@ -25,6 +25,7 @@ class SQLExplanationRequest(BaseModel):
     mdl_hash: Optional[str] = None
     thread_id: Optional[str] = None
     project_id: Optional[str] = None
+    user_id: Optional[str] = None
 
     @property
     def query_id(self) -> str:

diff --git a/wren-ai-service/src/web/v1/services/sql_regeneration.py b/wren-ai-service/src/web/v1/services/sql_regeneration.py
@@ -43,6 +43,7 @@ class SQLRegenerationRequest(BaseModel):
     mdl_hash: Optional[str] = None
     thread_id: Optional[str] = None
     project_id: Optional[str] = None
+    user_id: Optional[str] = None
 
     @property
     def query_id(self) -> str: