Kaynağa Gözat

fix(start): update FastAPI server port from 8100 to 8000

- Changed the server port in start.sh to ensure it runs on the standard FastAPI port (8000).

feat(pattern): modify _save_pattern_contents function to include pattern_name

- Updated the _save_pattern_contents function to accept pattern_name as a parameter for better content management.
- Adjusted the call to _save_pattern_contents in begin_pattern_task to pass the new parameter.
jihuaqiang 1 ay önce
ebeveyn
işleme
fa56562e4a
2 değiştirilmiş dosya ile 6 ekleme ve 4 silme
  1. 1 1
      start.sh
  2. 5 3
      tasks/pattern.py

+ 1 - 1
start.sh

@@ -25,5 +25,5 @@ if [ -f "requirements.txt" ]; then
 fi
 fi
 
 
 echo "Starting FastAPI server on http://localhost:8000 ..."
 echo "Starting FastAPI server on http://localhost:8000 ..."
-"$PYTHON" -m uvicorn main:app --host 0.0.0.0 --port 8100 --reload
+"$PYTHON" -m uvicorn main:app --host 0.0.0.0 --port 8000 --reload
 
 

+ 5 - 3
tasks/pattern.py

@@ -66,11 +66,12 @@ def _create_pattern_task(scene: SceneEnum, content_type: ContentTypeEnum) -> Opt
         return None
         return None
 
 
 
 
-def _save_pattern_contents(task_id: str, contents: List[ContentParam]) -> bool:
+def _save_pattern_contents(task_id: str, pattern_name: str, contents: List[ContentParam]) -> bool:
     """将聚类内容写入 workflow_pattern_task_content 表"""
     """将聚类内容写入 workflow_pattern_task_content 表"""
     sql = """
     sql = """
         INSERT INTO workflow_pattern_task_content (
         INSERT INTO workflow_pattern_task_content (
             task_id,
             task_id,
+            pattern_name,
             channel_content_id,
             channel_content_id,
             images,
             images,
             title,
             title,
@@ -79,7 +80,7 @@ def _save_pattern_contents(task_id: str, contents: List[ContentParam]) -> bool:
             body_text,
             body_text,
             video_url,
             video_url,
             weight_score
             weight_score
-        ) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)
+        ) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
     """
     """
 
 
     for content in contents:
     for content in contents:
@@ -87,6 +88,7 @@ def _save_pattern_contents(task_id: str, contents: List[ContentParam]) -> bool:
             images_str = json.dumps(content.images or []) if isinstance(content.images, list) else ""
             images_str = json.dumps(content.images or []) if isinstance(content.images, list) else ""
             params = (
             params = (
                 task_id,
                 task_id,
+                pattern_name,
                 content.channel_content_id,
                 content.channel_content_id,
                 images_str,
                 images_str,
                 content.title,
                 content.title,
@@ -181,7 +183,7 @@ def begin_pattern_task(param: PatternContentParam) -> Dict[str, Any]:
             )
             )
 
 
         # 3. 将内容写入 workflow_pattern_task_content 表
         # 3. 将内容写入 workflow_pattern_task_content 表
-        if not _save_pattern_contents(task.task_id, param.contents):
+        if not _save_pattern_contents(task.task_id, param.pattern_name, param.contents):
             return _build_error_response(
             return _build_error_response(
                 ERROR_CODE_FAILED,
                 ERROR_CODE_FAILED,
                 "写入聚类内容失败",
                 "写入聚类内容失败",