FECUOY commited on
Commit
2cd4955
·
verified ·
1 Parent(s): 4bec409

Update agents_config.py

Browse files
Files changed (1) hide show
  1. agents_config.py +77 -5
agents_config.py CHANGED
@@ -1,16 +1,16 @@
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" # تم التغيير هنا (I كبيرة)
14
  }
15
 
16
  def get_llm_config(model_id, user_token):
@@ -24,7 +24,79 @@ def get_llm_config(model_id, user_token):
24
  }
25
  ],
26
  "temperature": 0.7,
27
- "timeout": 300, # زيادة المهلة للنماذج الضخمة
28
  }
29
 
30
- # (بقية الكود الخاص بـ create_agents كما هو في الرد السابق)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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):
 
24
  }
25
  ],
26
  "temperature": 0.7,
27
+ "cache_seed": None,
28
  }
29
 
30
+ def create_agents(user_token):
31
+ """
32
+ هذه هي الدالة التي يبحث عنها ملف app.py
33
+ """
34
+ # إنشاء الوكلاء
35
+ analyst = autogen.AssistantAgent(
36
+ name="Analyst",
37
+ system_message="أنت المحلل. استخرج سياق الرواية والأحداث السابقة.",
38
+ llm_config=get_llm_config(MODEL_MAP["Analyst"], user_token)
39
+ )
40
+
41
+ style_guardian = autogen.AssistantAgent(
42
+ name="Style_Guardian",
43
+ system_message="أنت حارس الأسلوب. حافظ على أسلوب الكاتب الأصلي.",
44
+ llm_config=get_llm_config(MODEL_MAP["StyleGuardian"], user_token)
45
+ )
46
+
47
+ architect = autogen.AssistantAgent(
48
+ name="Architect",
49
+ system_message="أنت المهندس السردي. نظم الحبكة واقترح المسارات.",
50
+ llm_config=get_llm_config(MODEL_MAP["Architect"], user_token)
51
+ )
52
+
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
+ humanizer = autogen.AssistantAgent(
60
+ name="Humanizer",
61
+ system_message="أنت مضفي الطابع البشري. أضف الإبداع والحوار الطبيعي.",
62
+ llm_config=get_llm_config(MODEL_MAP["Humanizer"], user_token)
63
+ )
64
+
65
+ continuity_guard = autogen.AssistantAgent(
66
+ name="Continuity_Guard",
67
+ system_message="أنت حارس الاستمرارية. تأكد من التناسق الزمني.",
68
+ llm_config=get_llm_config(MODEL_MAP["ContinuityGuard"], user_token)
69
+ )
70
+
71
+ psychologist = autogen.AssistantAgent(
72
+ name="Psychologist",
73
+ system_message="أنت المدقق النفسي. حلل سلوك الشخصيات.",
74
+ llm_config=get_llm_config(MODEL_MAP["Psychologist"], user_token)
75
+ )
76
+
77
+ critic = autogen.AssistantAgent(
78
+ name="Critic",
79
+ system_message="أنت الناقد القاسي. قدم النقد النهائي.",
80
+ llm_config=get_llm_config(MODEL_MAP["Critic"], user_token)
81
+ )
82
+
83
+ editor = autogen.UserProxyAgent(
84
+ name="Editor_In_Chief",
85
+ human_input_mode="NEVER",
86
+ max_consecutive_auto_reply=10,
87
+ code_execution_config=False,
88
+ llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
89
+ system_message="أنت رئيس التحرير (InclusionAI/Ling-1T). وظيفتك دمج المخرجات في نص واحد متسق."
90
+ )
91
+
92
+ return {
93
+ "analyst": analyst,
94
+ "style_guardian": style_guardian,
95
+ "architect": architect,
96
+ "draft_writer": draft_writer,
97
+ "humanizer": humanizer,
98
+ "continuity_guard": continuity_guard,
99
+ "psychologist": psychologist,
100
+ "critic": critic,
101
+ "editor": editor
102
+ }