fix: output parse

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.0.5"
+version = "0.0.6"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

samples/quickstart-agent/main.py

@@ -16,6 +16,11 @@ class JokeEvent(Event):
     joke: str
 
 
+class CritiqueEvent(StopEvent):
+    joke: str
+    critique: str
+
+
 class JokeFlow(Workflow):
     llm = OpenAI()
 
@@ -28,12 +33,12 @@ async def generate_joke(self, ev: TopicEvent) -> JokeEvent:
         return JokeEvent(joke=str(response))
 
     @step
-    async def critique_joke(self, ev: JokeEvent) -> StopEvent:
+    async def critique_joke(self, ev: JokeEvent) -> CritiqueEvent:
         joke = ev.joke
 
         prompt = f"Give a thorough analysis and critique of the following joke: {joke}"
         response = await self.llm.acomplete(prompt)
-        return StopEvent(result=str(response))
+        return CritiqueEvent(joke=joke, critique=str(response))
 
 
 agent = JokeFlow(timeout=60, verbose=False)

samples/quickstart-agent/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "llama-quickstart-agent"
-version = "0.0.3"
+version = "0.0.5"
 description = "UiPath LlamaIndex Quickstart Agent"
 authors = [{ name = "John Doe", email = "john.doe@myemail.com" }]
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath-llamaindex>=0.0.5",
+    "uipath-llamaindex>=0.0.6",
     "llama-index-llms-openai>=0.2.2"
 ]

samples/quickstart-agent/uipath.json

@@ -2,7 +2,7 @@
   "entryPoints": [
     {
       "filePath": "agent",
-      "uniqueId": "f646d823-434c-4ef6-b8fb-6a239ee08974",
+      "uniqueId": "fef68d5e-56cd-4569-b18c-3bce0ef5343f",
       "type": "agent",
       "input": {
         "type": "object",
@@ -18,8 +18,20 @@
       },
       "output": {
         "type": "object",
-        "properties": {},
-        "required": []
+        "properties": {
+          "joke": {
+            "title": "Joke",
+            "type": "string"
+          },
+          "critique": {
+            "title": "Critique",
+            "type": "string"
+          }
+        },
+        "required": [
+          "joke",
+          "critique"
+        ]
       }
     }
   ],

samples/quickstart-agent/uv.lock

@@ -160,10 +160,10 @@ async def cleanup(self) -> None:
     def _serialize_object(self, obj):
         """Recursively serializes an object and all its nested components."""
         # Handle Pydantic models
-        if hasattr(obj, "dict"):
-            return self._serialize_object(obj.dict())
-        elif hasattr(obj, "model_dump"):
+        if hasattr(obj, "model_dump"):
             return self._serialize_object(obj.model_dump(by_alias=True))
+        elif hasattr(obj, "dict"):
+            return self._serialize_object(obj.dict())
         elif hasattr(obj, "to_dict"):
             return self._serialize_object(obj.to_dict())
         # Handle dictionaries
@@ -180,4 +180,4 @@ def _serialize_object(self, obj):
                 return obj
         # Return primitive types as is
         else:
-            return {"result": obj}
+            return obj