Merge pull request #132 from Chainlit/cd/ENG-1982

fix prompt printing

Author: constantinidan

literalai/observability/generation.py

@@ -13,10 +13,19 @@
 
 
 @unique
-class GenerationType(Enum):
+class GenerationType(str, Enum):
     CHAT = "CHAT"
     COMPLETION = "COMPLETION"
 
+    def __str__(self):
+        return self.value
+
+    def __repr__(self):
+        return f"GenerationType.{self.name}"
+
+    def to_json(self):
+        return self.value
+
 
 class GenerationMessage(TypedDict, total=False):
     uuid: Optional[str]

literalai/prompt_engineering/prompt.py

@@ -79,7 +79,6 @@ class Prompt(Utils):
     variables_default_values: Optional[Dict[str, Any]]
 
     def to_dict(self) -> PromptDict:
-        # Convert the Prompt instance to a dictionary matching the PromptDict structure
         return {
             "id": self.id,
             "createdAt": self.created_at,