youngtsai commited on
Commit
4c4564c
·
1 Parent(s): be1f3da

cutor_new_bot_structor init

Browse files
Files changed (1) hide show
  1. app.py +29 -0
app.py CHANGED
@@ -35,6 +35,7 @@ if is_env_local:
35
  CUTOR_OPEN_AI_KEY = config["CUTOR_OPEN_AI_KEY"]
36
  CUTOR_OPEN_AI_ASSISTANT_ID = config["CUTOR_OPEN_AI_ASSISTANT_ID"]
37
  CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID = config["CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID"]
 
38
  OPEN_AI_MODERATION_BOT1 = config["OPEN_AI_MODERATION_BOT1"]
39
 
40
  else:
@@ -43,6 +44,7 @@ else:
43
  CUTOR_OPEN_AI_KEY = os.getenv("CUTOR_OPEN_AI_KEY")
44
  CUTOR_OPEN_AI_ASSISTANT_ID = os.getenv("CUTOR_OPEN_AI_ASSISTANT_ID")
45
  CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID = os.getenv("CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID")
 
46
  OPEN_AI_MODERATION_BOT1 = os.getenv("OPEN_AI_MODERATION_BOT1", OPEN_AI_KEY)
47
 
48
  OPEN_AI_CLIENT = OpenAI(api_key=OPEN_AI_KEY)
@@ -1213,6 +1215,7 @@ def get_chinese_conversation_thread_id(thread_id):
1213
  return thread_id
1214
 
1215
  def get_chinese_paragraph_evaluate_content(thread_id, model, user_content, paragraph):
 
1216
  content = generate_content_by_open_ai_assistant(user_content, thread_id, model_name=model)
1217
  print(f"====generate_paragraph_evaluate====")
1218
  print(content)
@@ -1487,6 +1490,32 @@ def generate_content_by_open_ai_assistant_spelling_robot(paragraph, thread_id=No
1487
 
1488
  return response_text
1489
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1490
  # 小工具
1491
  def show_elements():
1492
  return gr.update(visible=True)
 
35
  CUTOR_OPEN_AI_KEY = config["CUTOR_OPEN_AI_KEY"]
36
  CUTOR_OPEN_AI_ASSISTANT_ID = config["CUTOR_OPEN_AI_ASSISTANT_ID"]
37
  CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID = config["CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID"]
38
+ CUTOR_OPEN_AI_STRUCTURE_ASSISTANT_ID = config["CUTOR_OPEN_AI_STRUCTURE_ASSISTANT_ID"]
39
  OPEN_AI_MODERATION_BOT1 = config["OPEN_AI_MODERATION_BOT1"]
40
 
41
  else:
 
44
  CUTOR_OPEN_AI_KEY = os.getenv("CUTOR_OPEN_AI_KEY")
45
  CUTOR_OPEN_AI_ASSISTANT_ID = os.getenv("CUTOR_OPEN_AI_ASSISTANT_ID")
46
  CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID = os.getenv("CUTOR_OPEN_AI_ASSISTANT_SPELLING_ID")
47
+ CUTOR_OPEN_AI_STRUCTURE_ASSISTANT_ID = os.getenv("CUTOR_OPEN_AI_STRUCTURE_ASSISTANT_ID")
48
  OPEN_AI_MODERATION_BOT1 = os.getenv("OPEN_AI_MODERATION_BOT1", OPEN_AI_KEY)
49
 
50
  OPEN_AI_CLIENT = OpenAI(api_key=OPEN_AI_KEY)
 
1215
  return thread_id
1216
 
1217
  def get_chinese_paragraph_evaluate_content(thread_id, model, user_content, paragraph):
1218
+ # 先做 主題與內容、遣詞造句
1219
  content = generate_content_by_open_ai_assistant(user_content, thread_id, model_name=model)
1220
  print(f"====generate_paragraph_evaluate====")
1221
  print(content)
 
1490
 
1491
  return response_text
1492
 
1493
+ # 結構 assistant id
1494
+ def generate_content_by_open_ai_assistant_structure_robot(paragraph, thread_id=None, model_name=None):
1495
+ verify_moderation(paragraph)
1496
+
1497
+ client = CUTOR_OPEN_AI_CLIENT
1498
+ assistant_id = CUTOR_OPEN_AI_STRUCTURE_ASSISTANT_ID
1499
+
1500
+ assistant = client.beta.assistants.update(
1501
+ assistant_id=assistant_id,
1502
+ tools=[{"type": "file_search"}],
1503
+ )
1504
+ print(f"My assistant: {assistant}")
1505
+ print(f"instructions: {assistant.instructions}")
1506
+
1507
+ try:
1508
+ thread_id = get_chinese_conversation_thread_id(thread_id) if not thread_id else thread_id
1509
+ thread = client.beta.threads.retrieve(thread_id)
1510
+ print(f"Thread ID: {thread.id}")
1511
+
1512
+ # Send the user message to the thread
1513
+ print("==============Send the user message to the thread====================")
1514
+ client.beta.threads.messages.create(thread_id=thread.id, role="user", content=user_content)
1515
+
1516
+
1517
+
1518
+
1519
  # 小工具
1520
  def show_elements():
1521
  return gr.update(visible=True)