Merge pull request #7 from UiPath/sample/quickstart_agent

samples: add quickstart agent

Author: cristipufu

samples/quickstart-agent/.env.example

@@ -0,0 +1 @@
+OPENAI_API_KEY=xxx

samples/quickstart-agent/llama_index.json

@@ -0,0 +1,7 @@
+{
+  "dependencies": ["."],
+  "workflows": {
+    "agentic_w": "main.py:agent"
+  },
+  "env": ".env"
+}

samples/quickstart-agent/main.py

@@ -0,0 +1,35 @@
+from llama_index.core.workflow import (
+    Event,
+    StartEvent,
+    StopEvent,
+    Workflow,
+    step,
+)
+from llama_index.llms.openai import OpenAI
+
+
+class JokeEvent(Event):
+    joke: str
+
+
+class JokeFlow(Workflow):
+    llm = OpenAI()
+
+    @step
+    async def generate_joke(self, ev: StartEvent) -> JokeEvent:
+        topic = ev.topic
+
+        prompt = f"Write your best joke about {topic}."
+        response = await self.llm.acomplete(prompt)
+        return JokeEvent(joke=str(response))
+
+    @step
+    async def critique_joke(self, ev: JokeEvent) -> StopEvent:
+        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))
+
+
+agent = JokeFlow(timeout=60, verbose=False)

samples/quickstart-agent/pyproject.toml

@@ -0,0 +1,11 @@
+[project]
+name = "llama-demo"
+version = "0.0.1"
+description = "UiPath LlamaIndex SDK"
+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.4",
+    "llama-index-llms-openai>=0.2.2"
+]