Small minor fixes to functions in Prompt functions (#1464)

Small minor fixes to functions in Prompt functions

Just a couple of minor fixes.

No tests exist for this so just gotta trust me I guess

python/src/aiconfig/schema.py

@@ -196,14 +196,18 @@ class Config:
         extra = "allow"
 
     def add_output(self, output: Output):
+        """
+        Add the output to the prompt's output list
+        """
+        if self.outputs is None:
+            self.outputs = [output]
         self.outputs.append(output)
 
-    def get_raw_prompt_from_config(self):
+    def get_raw_prompt_from_config(self) -> str:
         """Gets raw prompt from config"""
         if isinstance(self.input, str):
             return self.input
-        else:
-            return self.input.prompt
+        return str(self.input.data)
 
 
 class SchemaVersion(BaseModel):

python/tests/test_dataclass.py

@@ -18,8 +18,7 @@ def test_get_raw_prompt_from_prompt_object_prompt_input():
     prompt = Prompt(
         name="test",
         input=PromptInput(
-            prompt="This is a basic prompt",
-            data={},
+            data="This is a basic prompt",
         ),
         metadata=PromptMetadata(
             model=ModelMetadata(name="FakeMetadata", settings={}),