test graph draw

backend/business.py

@@ -227,49 +227,49 @@ def human_feedback(
     return history, acs     
 
 
-# if __name__ == "__main__":
-#     from IPython.display import Image,display
-#     abot = Agent(
-#         model=ChatOpenAI(model="gpt-4o", streaming=True),
-#         checkpointer=SqliteSaver.from_conn_string(":memory:"),
-#     )
-#     png_data = abot.graph.get_graph().draw_mermaid_png()
-#     with open('backend/business.png', 'wb') as file:
-#         file.write(png_data)
-
-#     thread = {"configurable": {"thread_id": "1"}}
-
-#     # print(abot.graph.get_state(thread))
-#     context = """Feature Config指的是一些特性的配置信息，例如应用程序的功能模块和设置。
-
-# 管理员可以通过界面进行新增、修改、分页查看、发布、撤销和删除操作；
-
-# 新增Feature Config应该提供 FeatureKey（通过FeatureKey 与Feature关联），title 和 description， data（配置的数据，数据格式依据Feature配置模版生成，由用户填写），saData（键值对埋点数据，由用户自定义输入），圈人条件（包括开始时间、截止时间、圈人白名单、地理位置、灰度比例等）
-
-# 新增后FeatureConfig记录状态默认为DRFAT。 管理课对DRAFT状态的数据执行发布操作，发布后记录变为PUBLISHED状态， 仅PUBLISHED状态的数据且符合圈人条件的配置C端可见。PUBLISHED状态的数据可执行撤销操作，撤销后记录变为DISABLED状态。
-
-# 对DRAFT状态的记录执行删除操作后，记录会从数据库中物理删除，非DRAFT状态的记录，执行删除操作，进行逻辑删除状态标记为DELETED，DELETED状态的数据在管理界面上不可见。"""
-#     story = """作为系统管理员
-# 我希望系统提供Feature Config新增/修改/分页查看/发布/撤销/删除的能力
-# 以便我增加/修改Feature中对应的配置信息"""
-#     for event in abot.graph.stream({"context": context.format(),
-#          "story": story.format(),
-#         "max_revisions": 10,
-#         "revision_number": 1}, thread, stream_mode="values"):
-#         print(event)
+if __name__ == "__main__":
+    from IPython.display import Image,display
+    abot = Agent(
+        model=ChatOpenAI(model="gpt-4o", streaming=True),
+        checkpointer=SqliteSaver.from_conn_string(":memory:"),
+    )
+    png_data = abot.graph.get_graph().draw_mermaid_png()
+    with open('backend/business.png', 'wb') as file:
+        file.write(png_data)
+
+    thread = {"configurable": {"thread_id": "1"}}
+
+    # print(abot.graph.get_state(thread))
+    context = """Feature Config指的是一些特性的配置信息，例如应用程序的功能模块和设置。
+
+管理员可以通过界面进行新增、修改、分页查看、发布、撤销和删除操作；
+
+新增Feature Config应该提供 FeatureKey（通过FeatureKey 与Feature关联），title 和 description， data（配置的数据，数据格式依据Feature配置模版生成，由用户填写），saData（键值对埋点数据，由用户自定义输入），圈人条件（包括开始时间、截止时间、圈人白名单、地理位置、灰度比例等）
+
+新增后FeatureConfig记录状态默认为DRFAT。 管理课对DRAFT状态的数据执行发布操作，发布后记录变为PUBLISHED状态， 仅PUBLISHED状态的数据且符合圈人条件的配置C端可见。PUBLISHED状态的数据可执行撤销操作，撤销后记录变为DISABLED状态。
+
+对DRAFT状态的记录执行删除操作后，记录会从数据库中物理删除，非DRAFT状态的记录，执行删除操作，进行逻辑删除状态标记为DELETED，DELETED状态的数据在管理界面上不可见。"""
+    story = """作为系统管理员
+我希望系统提供Feature Config新增/修改/分页查看/发布/撤销/删除的能力
+以便我增加/修改Feature中对应的配置信息"""
+    for event in abot.graph.stream({"context": context.format(),
+         "story": story.format(),
+        "max_revisions": 10,
+        "revision_number": 1}, thread, stream_mode="values"):
+        print(event)
     
 
-#     print(abot.graph.get_state(thread).values)
+    print(abot.graph.get_state(thread).values)
     
-#     for event in abot.graph.stream(None, thread, stream_mode="values"):
-#         print(event)
+    for event in abot.graph.stream(None, thread, stream_mode="values"):
+        print(event)
 
-#     print(abot.graph.get_state(thread).values)
+    print(abot.graph.get_state(thread).values)
 
-#     for event in abot.graph.stream(None, thread, stream_mode="values"):
-#         print(event)
+    for event in abot.graph.stream(None, thread, stream_mode="values"):
+        print(event)
 
-#     print(abot.graph.get_state(thread).values)
+    print(abot.graph.get_state(thread).values)
 
-#     # print(r, end= "\n\n")
-#     # print(r["question"], end="\n\n")
+    # print(r, end= "\n\n")
+    # print(r["question"], end="\n\n")

backend/essay_writer.py

@@ -189,29 +189,29 @@ def essay_writer_respond(
         
 
 
-# if __name__ == "__main__":
-#     from IPython.display import Image,display
-#     abot = Agent(
-#         model=ChatOpenAI(model="gpt-4o-mini", streaming=True),
-#         checkpointer=SqliteSaver.from_conn_string(":memory:"),
-#     )
-#     png_data = abot.graph.get_graph().draw_mermaid_png()
-#     display(Image(abot.graph.get_graph().draw_mermaid_png()))
-#     with open('backend/essay_writer.png', 'wb') as file:
-#         file.write(png_data)
-
-#     thread = {"configurable": {"thread_id": "1"}}
-
-#     print(abot.graph.get_state(thread))
-
-#     r = abot.graph.invoke(
-#         {"task": "Write an essay on the importance of education",
-#          "max_revisions": 2,
-#         "revision_number": 1,},
-#         thread,
-#     )
-
-#     print(abot.graph.get_state(thread).values['draft'])
+if __name__ == "__main__":
+    from IPython.display import Image,display
+    abot = Agent(
+        model=ChatOpenAI(model="gpt-4o-mini", streaming=True),
+        checkpointer=SqliteSaver.from_conn_string(":memory:"),
+    )
+    png_data = abot.graph.get_graph().draw_mermaid_png()
+    display(Image(abot.graph.get_graph().draw_mermaid_png()))
+    with open('backend/essay_writer.png', 'wb') as file:
+        file.write(png_data)
+
+    thread = {"configurable": {"thread_id": "1"}}
+
+    print(abot.graph.get_state(thread))
+
+    r = abot.graph.invoke(
+        {"task": "Write an essay on the importance of education",
+         "max_revisions": 2,
+        "revision_number": 1,},
+        thread,
+    )
+
+    print(abot.graph.get_state(thread).values['draft'])
     
-#     print(r, end= "\n\n")
-#     print(r["draft"], end="\n\n")
+    print(r, end= "\n\n")
+    print(r["draft"], end="\n\n")