fix(writer): smodin开关

writer/writer.py

@@ -65,20 +65,23 @@ async def designer(state: GraphState):
                 break
 
     response = await designer_model.ainvoke({"messages": [("user", f"{messages[::-1]}")]})
-    payload = {"language": "en",
-               "text": response.content,
-               "strength": 4,
-               }
-    async with aiohttp.ClientSession() as session:
-        try:
-            async with session.post(os.environ.get("SMODIN_BASE_URL"), headers=SMODIN_HEADERS, json=payload) as res:
-                rewritten_text = await res.json()
-        except:
-            # print("Error:", res.status, await res.text())
-            return {"messages": []}
-        finally:
-            rewritten_text = await no_conclusion_model.ainvoke(
-                {"messages": [("user", rewritten_text["rewrites"][0]["rewrite"])]})
+    if os.environ.get("USE_SMODIN") == 'true':
+        payload = {"language": "en",
+                   "text": response.content,
+                   "strength": 4,
+                   }
+        async with aiohttp.ClientSession() as session:
+            try:
+                async with session.post(os.environ.get("SMODIN_BASE_URL"), headers=SMODIN_HEADERS, json=payload) as res:
+                    rewritten_text = await res.json()
+            except:
+                # print("Error:", res.status, await res.text())
+                return {"messages": []}
+            finally:
+                rewritten_text = await no_conclusion_model.ainvoke(
+                    {"messages": [("user", rewritten_text["rewrites"][0]["rewrite"])]})
+    else:
+        rewritten_text = response
     i = state["time"]
     return {"messages": [rewritten_text.content], "time": i + 1}
     # return {"messages": [response.content], "time": i + 1}