Extend input_personas

shared.py

@@ -38,9 +38,10 @@ class ModelConfig(BaseModel):
         
 
 class Client:
-    def __init__(self, api_url, api_key):
+    def __init__(self, api_url, api_key, personas):
         self.api_url = api_url
         self.api_key = api_key
+        self.input_personas = personas
         
         self.init_all()
 
@@ -72,8 +73,9 @@ class Client:
                                     subfolder="datasets",
                                     revision=self.revision)
             self.config = ModelConfig.from_yaml(config_path)
-            self.personas = self.config.pile.persona2system
+            personas = self.config.pile.persona2system
         except EntryNotFoundError:
-            self.personas = {}
+            personas = {}
 
-        self.personas["default"] = None
+        personas["default"] = None
+        self.personas = self.input_personas | personas