FECUOY commited on
Commit
a11de4b
·
verified ·
1 Parent(s): 7a54cf4

Update agents_config.py

Browse files
Files changed (1) hide show
  1. agents_config.py +33 -84
agents_config.py CHANGED
@@ -1,16 +1,16 @@
1
  import autogen
2
 
3
- # القائمة المحدثة بالنماذج مع لاحقة featherless-ai
4
  MODEL_MAP = {
5
- "Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct:featherless-ai",
6
- "StyleGuardian": "openai/gpt-oss-120b:featherless-ai",
7
- "Architect": "deepseek-ai/DeepSeek-V3.2-Exp:featherless-ai",
8
- "DraftWriter": "deepcogito/cogito-v2-preview-llama-405B:featherless-ai",
9
- "Humanizer": "deepcogito/cogito-671b-v2.1:featherless-ai",
10
- "ContinuityGuard": "meta-llama/Llama-3.3-70B-Instruct:featherless-ai",
11
- "Psychologist": "zai-org/GLM-4.7-FP8:featherless-ai",
12
- "Critic": "Qwen/Qwen3-235B-A22B-Thinking-2507:featherless-ai",
13
- "Editor": "inclusionAI/Ling-1T:featherless-ai"
14
  }
15
 
16
  def get_llm_config(model_id, user_token):
@@ -28,84 +28,33 @@ def get_llm_config(model_id, user_token):
28
  }
29
 
30
  def create_agents(user_token):
31
- # 1. وكيل التحليل
32
- analyst = autogen.AssistantAgent(
33
- name="Analyst",
34
- system_message="أنت المحلل. قم بتفكيك النص المرفوع واستخراج الشخصيات والدوافع والبيئة الزمانية والمكانية.",
35
- llm_config=get_llm_config(MODEL_MAP["Analyst"], user_token)
36
- )
37
-
38
- # 2. وكيل الأسلوب
39
- style_guardian = autogen.AssistantAgent(
40
- name="Style_Guardian",
41
- system_message="أنت حارس الأسلوب. حلل نبرة الكاتب في الملف المرفوع وتأكد أن الإكمال يتبع نفس الإيقاع اللغوي.",
42
- llm_config=get_llm_config(MODEL_MAP["StyleGuardian"], user_token)
43
- )
44
-
45
- # 3. المهندس السردي
46
- architect = autogen.AssistantAgent(
47
- name="Architect",
48
- system_message="أنت المهندس السردي. اقترح هيكل الأحداث للفصل القادم بناءً على التحليل.",
49
- llm_config=get_llm_config(MODEL_MAP["Architect"], user_token)
50
- )
51
-
52
- # 4. كاتب المسودة
53
- draft_writer = autogen.AssistantAgent(
54
- name="Draft_Writer",
55
- system_message="أنت كاتب المسودة. قم بكتابة الأحداث بناءً على خطة المهندس ومعايير حارس الأسلوب.",
56
- llm_config=get_llm_config(MODEL_MAP["DraftWriter"], user_token)
57
- )
58
-
59
- # 5. وكيل الأنسنة
60
- humanizer = autogen.AssistantAgent(
61
- name="Humanizer",
62
- system_message="أنت وكيل الأنسنة. أضف تفاصيل حسية، مشاعر عميقة، وحوارات واقعية تجعل النص ينبض بالحياة.",
63
- llm_config=get_llm_config(MODEL_MAP["Humanizer"], user_token)
64
- )
65
-
66
- # 6. حارس الاستمرارية
67
- continuity_guard = autogen.AssistantAgent(
68
- name="Continuity_Guard",
69
- system_message="أنت حارس الاستمرارية. تأكد من عدم وجود تناقضات مع الأحداث الواردة في الملف الأصلي.",
70
- llm_config=get_llm_config(MODEL_MAP["ContinuityGuard"], user_token)
71
- )
72
-
73
- # 7. الأخصائي النفسي
74
- psychologist = autogen.AssistantAgent(
75
- name="Psychologist",
76
- system_message="أنت الأخصائي النفسي. تأكد من أن تصرفات الشخصيات في الإكمال تتوافق مع تركيبتهم النفسية.",
77
- llm_config=get_llm_config(MODEL_MAP["Psychologist"], user_token)
78
- )
79
-
80
- # 8. الناقد
81
- critic = autogen.AssistantAgent(
82
- name="Critic",
83
- system_message="أنت الناقد. راجع المخرج النهائي بصرامة واقترح تعديلات لتحسين الجودة الأدبية.",
84
- llm_config=get_llm_config(MODEL_MAP["Critic"], user_token)
85
- )
86
 
87
- # 9. رئيس التحرير - تم تعديل human_input_mode إلى NEVER
88
- editor = autogen.UserProxyAgent(
89
  name="Editor_In_Chief",
90
- human_input_mode="NEVER",
91
  max_consecutive_auto_reply=15,
92
  code_execution_config=False,
93
  llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
94
- system_message="""أنت رئيس التحرير (Ling-1T). وظيفتك هي:
95
- 1. استلام النص من الملف المرفوع وتوزيعه على الوكلاء.
96
- 2. جمع المخرجات وصهرها في 'مخطوطة نهائية' منظمة.
97
- 3. يجب أن يكون مخرجك النهائي منسقاً كالتالي: [العنوان] -> [رقم الفصل] -> [المحتوى الروائي].
98
- أنت المسؤول عن الجودة النهائية للملف الذي سيتم تحميله."""
99
  )
100
 
101
- return {
102
- "analyst": analyst,
103
- "style_guardian": style_guardian,
104
- "architect": architect,
105
- "draft_writer": draft_writer,
106
- "humanizer": humanizer,
107
- "continuity_guard": continuity_guard,
108
- "psychologist": psychologist,
109
- "critic": critic,
110
- "editor": editor
111
- }
 
1
  import autogen
2
 
3
+ # النماذج التي أكدت عملها بدون إضافات المزود اليدوية
4
  MODEL_MAP = {
5
+ "Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
6
+ "StyleGuardian": "openai/gpt-oss-120b",
7
+ "Architect": "deepseek-ai/DeepSeek-V3.2-Exp",
8
+ "DraftWriter": "deepcogito/cogito-v2-preview-llama-405B",
9
+ "Humanizer": "deepcogito/cogito-671b-v2.1",
10
+ "ContinuityGuard": "meta-llama/Llama-3.3-70B-Instruct",
11
+ "Psychologist": "zai-org/GLM-4.7-FP8",
12
+ "Critic": "Qwen/Qwen3-235B-A22B-Thinking-2507",
13
+ "Editor": "inclusionAI/Ling-1T"
14
  }
15
 
16
  def get_llm_config(model_id, user_token):
 
28
  }
29
 
30
  def create_agents(user_token):
31
+ # إنشاء الوكلاء المساعدين
32
+ def create_asst(name, msg, model_key):
33
+ return autogen.AssistantAgent(
34
+ name=name,
35
+ system_message=msg,
36
+ llm_config=get_llm_config(MODEL_MAP[model_key], user_token)
37
+ )
38
+
39
+ agents = {
40
+ "analyst": create_asst("Analyst", "أنت المحلل البنيوي للنص.", "Analyst"),
41
+ "style_guardian": create_asst("Style_Guardian", "أنت حارس الأسلوب الأدبي.", "StyleGuardian"),
42
+ "architect": create_asst("Architect", "أنت مهندس الحبكة السردية.", "Architect"),
43
+ "draft_writer": create_asst("Draft_Writer", "أنت كاتب المسودة الأولية.", "DraftWriter"),
44
+ "humanizer": create_asst("Humanizer", "أنت خبير الأنسنة والتفاصيل الحسية.", "Humanizer"),
45
+ "continuity_guard": create_asst("Continuity_Guard", "أنت مراقب الاستمرارية ومنطق الأحداث.", "ContinuityGuard"),
46
+ "psychologist": create_asst("Psychologist", "أنت محلل الدوافع النفسية للشخصيات.", "Psychologist"),
47
+ "critic": create_asst("Critic", "أنت الناقد الأدبي الصارم.", "Critic"),
48
+ }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
49
 
50
+ # رئيس التحرير بوضع NEVER للعمل مع واجهة الويب
51
+ agents["editor"] = autogen.UserProxyAgent(
52
  name="Editor_In_Chief",
53
+ human_input_mode="NEVER",
54
  max_consecutive_auto_reply=15,
55
  code_execution_config=False,
56
  llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
57
+ system_message="أنت رئيس التحرير المسؤول عن دمج كل مساهمات الوكلاء في نص روائي واحد متكامل."
 
 
 
 
58
  )
59
 
60
+ return agents