|
47 | 47 | "metadata": {}, |
48 | 48 | "outputs": [], |
49 | 49 | "source": [ |
50 | | - "import asyncio\n", |
51 | | - "\n", |
52 | | - "from dotenv import load_dotenv\n", |
53 | 50 | "import nest_asyncio\n", |
| 51 | + "from ag_ui.core import (\n", |
| 52 | + " AssistantMessage,\n", |
| 53 | + " MessagesSnapshotEvent,\n", |
| 54 | + " TextMessageChunkEvent,\n", |
| 55 | + " UserMessage,\n", |
| 56 | + ")\n", |
| 57 | + "from dotenv import load_dotenv\n", |
54 | 58 | "from IPython.display import display\n", |
55 | 59 | "from langchain_openai import ChatOpenAI\n", |
56 | 60 | "\n", |
57 | | - "from ragas.dataset_schema import EvaluationDataset, SingleTurnSample, MultiTurnSample\n", |
| 61 | + "from ragas.dataset_schema import EvaluationDataset, MultiTurnSample, SingleTurnSample\n", |
58 | 62 | "from ragas.integrations.ag_ui import (\n", |
59 | | - " evaluate_ag_ui_agent,\n", |
60 | | - " convert_to_ragas_messages,\n", |
61 | 63 | " convert_messages_snapshot,\n", |
| 64 | + " convert_to_ragas_messages,\n", |
| 65 | + " evaluate_ag_ui_agent,\n", |
62 | 66 | ")\n", |
| 67 | + "from ragas.llms import LangchainLLMWrapper\n", |
63 | 68 | "from ragas.messages import HumanMessage, ToolCall\n", |
64 | 69 | "from ragas.metrics import FactualCorrectness, ToolCallF1\n", |
65 | | - "from ragas.llms import LangchainLLMWrapper\n", |
66 | | - "from ag_ui.core import (\n", |
67 | | - " MessagesSnapshotEvent,\n", |
68 | | - " TextMessageChunkEvent,\n", |
69 | | - " UserMessage,\n", |
70 | | - " AssistantMessage,\n", |
71 | | - ")\n", |
72 | 70 | "\n", |
73 | 71 | "load_dotenv()\n", |
74 | 72 | "# Patch the existing notebook loop so we can await coroutines safely\n", |
|
0 commit comments