Parcourir la source

Update prompt_utils: add format staff profile v2

StrayWarrior il y a 2 jours
Parent
commit
841b11ebea
1 fichiers modifiés avec 9 ajouts et 0 suppressions
  1. 9 0
      pqai_agent/utils/prompt_utils.py

+ 9 - 0
pqai_agent/utils/prompt_utils.py

@@ -1,5 +1,9 @@
+import json
+from io import StringIO
 from typing import Dict
 
+import yaml
+
 
 def format_agent_profile(profile: Dict) -> str:
     fields = [
@@ -22,6 +26,11 @@ def format_agent_profile(profile: Dict) -> str:
         strings_to_join.append(cur_string)
     return "\n".join(strings_to_join)
 
+def format_agent_profile_v2(profile: Dict) -> str:
+    str_stream = StringIO()
+    yaml.dump(profile, str_stream, indent=2, allow_unicode=True)
+    return str_stream.getvalue()
+
 def format_user_profile(profile: Dict) -> str:
     """
     :param profile: