FECUOY commited on
Commit
f25bb93
·
verified ·
1 Parent(s): e6e65b5

Update agents_config.py

Browse files
Files changed (1) hide show
  1. agents_config.py +31 -20
agents_config.py CHANGED
@@ -1,6 +1,6 @@
1
  import autogen
2
 
3
- # القائمة المباشرة للنماذج
4
  MODEL_MAP = {
5
  "Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
6
  "StyleGuardian": "openai/gpt-oss-120b",
@@ -14,42 +14,53 @@ MODEL_MAP = {
14
  }
15
 
16
  def get_llm_config(model_id, user_token):
 
17
  return {
18
- "config_list": [{
19
- "model": model_id,
20
- "api_key": user_token,
21
- "base_url": "https://router.huggingface.co/v1",
22
- "api_type": "openai",
23
- }],
 
 
24
  "temperature": 0.7,
25
- "cache_seed": None,
26
  }
27
 
28
  def create_agents(user_token):
29
- def mk_asst(name, msg, model_key):
 
 
 
30
  return autogen.AssistantAgent(
31
  name=name,
32
- system_message=msg,
33
  llm_config=get_llm_config(MODEL_MAP[model_key], user_token)
34
  )
35
 
36
  agents = {
37
- "analyst": mk_asst("Analyst", "محلل نصوص.", "Analyst"),
38
- "style_guardian": mk_asst("Style_Guardian", "خبير أسلوب.", "StyleGuardian"),
39
- "architect": mk_asst("Architect", "مهندس حبكة.", "Architect"),
40
- "draft_writer": mk_asst("Draft_Writer", "كاتب مسودة.", "DraftWriter"),
41
- "humanizer": mk_asst("Humanizer", "خبير مشاعر.", "Humanizer"),
42
- "continuity_guard": mk_asst("Continuity_Guard", "مراقب استمرارية.", "ContinuityGuard"),
43
- "psychologist": mk_asst("Psychologist", "محلل نفسي.", "Psychologist"),
44
- "critic": mk_asst("Critic", "ناقد أدبي.", "Critic"),
45
  }
46
 
 
 
47
  agents["editor"] = autogen.UserProxyAgent(
48
  name="Editor_In_Chief",
49
- human_input_mode="NEVER",
50
  max_consecutive_auto_reply=10,
51
  code_execution_config=False,
52
  llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
53
- system_message="أنت رئيس التحرير، اجمع النتائج في نص واحد."
 
 
54
  )
 
55
  return agents
 
1
  import autogen
2
 
3
+ # خريطة النماذج المباشرة الموزعة على الوكلاء
4
  MODEL_MAP = {
5
  "Analyst": "Qwen/Qwen3-Next-80B-A3B-Instruct",
6
  "StyleGuardian": "openai/gpt-oss-120b",
 
14
  }
15
 
16
  def get_llm_config(model_id, user_token):
17
+ """إنشاء إعدادات الاتصال بالنموذج عبر راوتر Hugging Face"""
18
  return {
19
+ "config_list": [
20
+ {
21
+ "model": model_id,
22
+ "api_key": user_token,
23
+ "base_url": "https://router.huggingface.co/v1",
24
+ "api_type": "openai",
25
+ }
26
+ ],
27
  "temperature": 0.7,
28
+ "cache_seed": None, # لضمان نتائج متجددة دائماً
29
  }
30
 
31
  def create_agents(user_token):
32
+ """توليد فريق الوكلاء وتوزيع المهام عليهم"""
33
+
34
+ # وظيفة داخلية لتبسيط إنشاء الوكلاء المساعدين
35
+ def mk_asst(name, sys_msg, model_key):
36
  return autogen.AssistantAgent(
37
  name=name,
38
+ system_message=sys_msg,
39
  llm_config=get_llm_config(MODEL_MAP[model_key], user_token)
40
  )
41
 
42
  agents = {
43
+ "analyst": mk_asst("Analyst", "أنت المحلل البنيوي للنص، فكك الأفكار وحدد الروابط العمقية.", "Analyst"),
44
+ "style_guardian": mk_asst("Style_Guardian", "أنت حارس الأسلوب الأدبي، تأكد من جودة اللغة والجماليات السردية.", "StyleGuardian"),
45
+ "architect": mk_asst("Architect", "أنت مهندس الحبكة، قم ببناء تسلسل أحداث متماسك ومشوق.", "Architect"),
46
+ "draft_writer": mk_asst("Draft_Writer", "أنت كاتب المسودة الأولية، حول الأفكار إلى نص سردي روائي.", "DraftWriter"),
47
+ "humanizer": mk_asst("Humanizer", "أنت خبير الأنسنة، بث المشاعر في الشخصيات واجعل الحوارات نابضة بالحياة.", "Humanizer"),
48
+ "continuity_guard": mk_asst("Continuity_Guard", "أنت مراقب الاستمرارية، امنع التناقضات في الأحداث أو صفات الشخصيات.", "ContinuityGuard"),
49
+ "psychologist": mk_asst("Psychologist", "أنت محلل الدوافع النفسية، اجعل تصرفات الشخصيات مبررة نفسياً.", "Psychologist"),
50
+ "critic": mk_asst("Critic", "أنت الناقد الأدبي، راجع النص النهائي وقدم ملاحظات قاسية للتحسين.", "Critic"),
51
  }
52
 
53
+ # رئيس التحرير: المسؤول عن الإدارة والصياغة النهائية
54
+ # تم ضبط human_input_mode إلى NEVER لضمان العمل التلقائي في بيئة الويب
55
  agents["editor"] = autogen.UserProxyAgent(
56
  name="Editor_In_Chief",
57
+ human_input_mode="NEVER",
58
  max_consecutive_auto_reply=10,
59
  code_execution_config=False,
60
  llm_config=get_llm_config(MODEL_MAP["Editor"], user_token),
61
+ system_message="""أنت رئيس التحرير (Ling-1T).
62
+ مهمتك هي قيادة نقاش الوكلاء ودمج مخرجاتهم لصياغة الفصل الروائي النهائي بأسلوب أدبي رفيع.
63
+ يجب أن تنتهي المحادثة دائماً بالنص الكامل للمخطوطة."""
64
  )
65
+
66
  return agents