Add Jelly

Signed-off-by: Choonho Son <[email protected]>

requirements.txt

@@ -0,0 +1,2 @@
+llama-cpp-python
+langchain

src/jellys/Belty/CHARACTER.yml

@@ -0,0 +1,5 @@
+---
+name: Belty
+role: Stock Analyst
+
+

src/main.py

@@ -1,10 +1,33 @@
 # This is main function
 import os
+import persona
 
-persona = os.environ.get('PERSONA')
+persona_name = os.environ.get('PERSONA', 'Wonny')
+
+def _list_all_jellys():
+    directory_path = './jellys'
+    contents = os.listdir(directory_path)
+    result = []
+    for item in contents:
+        result.append(item)
+    return result
+
+def _run_jelly(persona_name):
+    try:
+        print(f"Run {persona_name} jelly...")
+        persona.main(persona_name)
+    except Exception as e:
+        print(e)
+
+def main(persona_name):
+    if persona_name == 'Wonny':
+        jellys = _list_all_jellys()
+    else:
+        jellys = [persona_name]
+
+    for jelly in jellys:
+        _run_jelly(jelly)
 
-def main(persona):
-    print(f"Run {persona} function...")
 
 if __name__ == "__main__":
-    main(persona)
+    main(persona_name)

src/persona.py

@@ -0,0 +1,16 @@
+# Read name
+
+import yaml
+
+persona_file = "CHARACTER.yml"
+
+def load_persona(name):
+    persona_path = f"./jellys/{name}/{persona_file}"
+    data = yaml.safe_load(open(persona_path))
+    print(data)
+
+def main(name):
+    load_persona(name)
+
+if __name__ == "__main__":
+    main("Belty")