Fix ag-ui example
This commit is contained in:
Yiorgis Gozadinos
parent
99176ba0a1
commit
110e020122
5 changed files with 64 additions and 32 deletions
|
|
@ -9,7 +9,7 @@ dependencies = [
|
|||
"uvicorn[standard]>=0.34.2",
|
||||
"pydantic-ai-slim[ag-ui,openai]>=1.17.0",
|
||||
"python-dotenv>=1.0.1",
|
||||
"haiku-rag-slim @ file:///Users/ggozad/dev/open-source/haiku.rag-agui/haiku_rag_slim",
|
||||
"haiku.rag-slim[agui]>=0.20.0",
|
||||
]
|
||||
|
||||
[dependency-groups]
|
||||
|
|
|
|||
|
|
@ -25,12 +25,22 @@ interface GapRecord {
|
|||
resolved_by: string[];
|
||||
}
|
||||
|
||||
interface Citation {
|
||||
document_id: string;
|
||||
chunk_id: string;
|
||||
document_uri: string;
|
||||
document_title?: string;
|
||||
page_numbers: number[];
|
||||
headings?: string[];
|
||||
content: string;
|
||||
}
|
||||
|
||||
interface SearchAnswer {
|
||||
query: string;
|
||||
answer: string;
|
||||
confidence: number;
|
||||
context: string[];
|
||||
sources: string[];
|
||||
cited_chunks: string[];
|
||||
citations: Citation[];
|
||||
}
|
||||
|
||||
interface ResearchContext {
|
||||
|
|
@ -94,7 +104,6 @@ function AgentContent() {
|
|||
},
|
||||
});
|
||||
|
||||
|
||||
return (
|
||||
<>
|
||||
<style>{`
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
"use client";
|
||||
|
||||
import { Markdown } from "@copilotkit/react-ui";
|
||||
import { useState, useCallback } from "react";
|
||||
import { useCallback, useState } from "react";
|
||||
|
||||
interface VisualGroundingState {
|
||||
isOpen: boolean;
|
||||
|
|
@ -127,7 +127,7 @@ export default function StateDisplay({ state }: StateDisplayProps) {
|
|||
|
||||
try {
|
||||
const response = await fetch(
|
||||
`${process.env.NEXT_PUBLIC_API_URL || "http://localhost:8000"}/api/visualize/${chunkId}`
|
||||
`${process.env.NEXT_PUBLIC_API_URL || "http://localhost:8000"}/api/visualize/${chunkId}`,
|
||||
);
|
||||
const data = await response.json();
|
||||
|
||||
|
|
@ -241,14 +241,25 @@ export default function StateDisplay({ state }: StateDisplayProps) {
|
|||
marginBottom: state.iterations > 0 ? "1rem" : 0,
|
||||
}}
|
||||
>
|
||||
<div style={{ display: "flex", alignItems: "center", gap: "0.5rem" }}>
|
||||
<div
|
||||
style={{ display: "flex", alignItems: "center", gap: "0.5rem" }}
|
||||
>
|
||||
<span>⏳</span>
|
||||
<span style={{ fontWeight: "600", color: "#2b6cb0" }}>
|
||||
{state.current_activity.replace(/_/g, " ").replace(/\b\w/g, (c) => c.toUpperCase())}
|
||||
{state.current_activity
|
||||
.replace(/_/g, " ")
|
||||
.replace(/\b\w/g, (c) => c.toUpperCase())}
|
||||
</span>
|
||||
</div>
|
||||
{state.current_activity_message && (
|
||||
<div style={{ fontSize: "0.875rem", color: "#4299e1", marginTop: "0.25rem", marginLeft: "1.5rem" }}>
|
||||
<div
|
||||
style={{
|
||||
fontSize: "0.875rem",
|
||||
color: "#4299e1",
|
||||
marginTop: "0.25rem",
|
||||
marginLeft: "1.5rem",
|
||||
}}
|
||||
>
|
||||
{state.current_activity_message}
|
||||
</div>
|
||||
)}
|
||||
|
|
@ -449,7 +460,9 @@ export default function StateDisplay({ state }: StateDisplayProps) {
|
|||
>
|
||||
⏳
|
||||
</div>
|
||||
<div style={{ flex: 1, fontSize: "0.875rem", color: "#4a5568" }}>
|
||||
<div
|
||||
style={{ flex: 1, fontSize: "0.875rem", color: "#4a5568" }}
|
||||
>
|
||||
<Markdown content={question} />
|
||||
</div>
|
||||
</div>
|
||||
|
|
@ -510,7 +523,8 @@ export default function StateDisplay({ state }: StateDisplayProps) {
|
|||
marginTop: "0.25rem",
|
||||
}}
|
||||
>
|
||||
Confidence: {(qaResponse.confidence * 100).toFixed(0)}%
|
||||
Confidence: {(qaResponse.confidence * 100).toFixed(0)}
|
||||
%
|
||||
</div>
|
||||
</div>
|
||||
<span
|
||||
|
|
@ -648,7 +662,9 @@ export default function StateDisplay({ state }: StateDisplayProps) {
|
|||
</div>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => fetchVisualGrounding(citation.chunk_id)}
|
||||
onClick={() =>
|
||||
fetchVisualGrounding(citation.chunk_id)
|
||||
}
|
||||
style={{
|
||||
marginTop: "0.5rem",
|
||||
padding: "0.25rem 0.5rem",
|
||||
|
|
|
|||
45
examples/ag-ui-research/frontend/package-lock.json
generated
45
examples/ag-ui-research/frontend/package-lock.json
generated
|
|
@ -8,7 +8,7 @@
|
|||
"name": "ag-ui-frontend",
|
||||
"version": "0.1.0",
|
||||
"dependencies": {
|
||||
"@ag-ui/client": "^0.0.40",
|
||||
"@ag-ui/client": "^0.0.42",
|
||||
"@copilotkit/react-core": "^1.10.6",
|
||||
"@copilotkit/react-ui": "^1.10.6",
|
||||
"@copilotkit/runtime": "^1.10.6",
|
||||
|
|
@ -39,15 +39,16 @@
|
|||
}
|
||||
},
|
||||
"node_modules/@ag-ui/client": {
|
||||
"version": "0.0.40",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/client/-/client-0.0.40.tgz",
|
||||
"integrity": "sha512-4ftyZgMN7DIAX64k7Mdex/KGq7lfz8yxEKzniqosD6TE/xk65k4Z0v3bxTzPk2iS2+Cj2uVBgFkb5lC7k5Loqg==",
|
||||
"version": "0.0.42",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/client/-/client-0.0.42.tgz",
|
||||
"integrity": "sha512-zAbP+sZJImR5bUpR2ni7RtuuNZMuesaxviynyIgzKlr1k2VCM49mFpbDUKU4TH4Cneu+Xe7OEnO8qCOCIzBAww==",
|
||||
"peer": true,
|
||||
"dependencies": {
|
||||
"@ag-ui/core": "0.0.39",
|
||||
"@ag-ui/encoder": "0.0.39",
|
||||
"@ag-ui/proto": "0.0.39",
|
||||
"@ag-ui/core": "0.0.42",
|
||||
"@ag-ui/encoder": "0.0.42",
|
||||
"@ag-ui/proto": "0.0.42",
|
||||
"@types/uuid": "^10.0.0",
|
||||
"compare-versions": "^6.1.1",
|
||||
"fast-json-patch": "^3.1.1",
|
||||
"rxjs": "7.8.1",
|
||||
"untruncate-json": "^0.0.1",
|
||||
|
|
@ -56,9 +57,9 @@
|
|||
}
|
||||
},
|
||||
"node_modules/@ag-ui/core": {
|
||||
"version": "0.0.39",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/core/-/core-0.0.39.tgz",
|
||||
"integrity": "sha512-T5Hp4oFkQ+H5MynWAvSwrX/rNYJOD+PJ4qPQ0o771oSZQAxoIvDDft47Cx5wRyBNNLXAe1RWqJjfWUUwJFNKqA==",
|
||||
"version": "0.0.42",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/core/-/core-0.0.42.tgz",
|
||||
"integrity": "sha512-C2hMg4Gs5oiUDgK9cA2RsTwSSmFZdIsqPklDrFw/Ue+quH6EU3vKp5YoOq7nuaQYO4pO8Em+Z+l5/M5PpcvP1g==",
|
||||
"peer": true,
|
||||
"dependencies": {
|
||||
"rxjs": "7.8.1",
|
||||
|
|
@ -66,13 +67,13 @@
|
|||
}
|
||||
},
|
||||
"node_modules/@ag-ui/encoder": {
|
||||
"version": "0.0.39",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/encoder/-/encoder-0.0.39.tgz",
|
||||
"integrity": "sha512-6fsoFwPWkStK7Uyj3pwBn7+aQjUWf7pbDTSI43cD53sBLvTr5oEFNnoKOzRfC5UqvHc4JjUIuLKPQyjHRwWg4g==",
|
||||
"version": "0.0.42",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/encoder/-/encoder-0.0.42.tgz",
|
||||
"integrity": "sha512-97B5MMCSs82t/y41uk2NrLBYFhbvn4kYsKQHMCfy8tjSWubyxh3zP7N9yHo8zJeSPe3WvzTvclyXNiGxSOsorg==",
|
||||
"peer": true,
|
||||
"dependencies": {
|
||||
"@ag-ui/core": "0.0.39",
|
||||
"@ag-ui/proto": "0.0.39"
|
||||
"@ag-ui/core": "0.0.42",
|
||||
"@ag-ui/proto": "0.0.42"
|
||||
}
|
||||
},
|
||||
"node_modules/@ag-ui/langgraph": {
|
||||
|
|
@ -92,12 +93,12 @@
|
|||
}
|
||||
},
|
||||
"node_modules/@ag-ui/proto": {
|
||||
"version": "0.0.39",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/proto/-/proto-0.0.39.tgz",
|
||||
"integrity": "sha512-xlj/PzZHkJ3CgoQC5QP9g7DEl/78wUK1+A2rdkoLKoNAMOkM2g6jKw0N88iFIh5GZhtiCNN2wb8XwRWPYx9XQQ==",
|
||||
"version": "0.0.42",
|
||||
"resolved": "https://registry.npmjs.org/@ag-ui/proto/-/proto-0.0.42.tgz",
|
||||
"integrity": "sha512-NDUwSgMnGEqxZGkWIJ1ge5t3Q7Kiddj360x2JAWaIfv9w+7tDJ0pmgyzf3/SXp605aY2wZiDLBtJ6jKZeg1lFg==",
|
||||
"peer": true,
|
||||
"dependencies": {
|
||||
"@ag-ui/core": "0.0.39",
|
||||
"@ag-ui/core": "0.0.42",
|
||||
"@bufbuild/protobuf": "^2.2.5",
|
||||
"@protobuf-ts/protoc": "^2.11.1"
|
||||
}
|
||||
|
|
@ -5266,6 +5267,12 @@
|
|||
"node": ">= 12"
|
||||
}
|
||||
},
|
||||
"node_modules/compare-versions": {
|
||||
"version": "6.1.1",
|
||||
"resolved": "https://registry.npmjs.org/compare-versions/-/compare-versions-6.1.1.tgz",
|
||||
"integrity": "sha512-4hm4VPpIecmlg59CHXnRDnqGplJFrbLG4aFEl5vl6cK1u76ws3LLvX7ikFnTDl5vo39sjWD6AaDPYodJp/NNHg==",
|
||||
"license": "MIT"
|
||||
},
|
||||
"node_modules/console-table-printer": {
|
||||
"version": "2.14.6",
|
||||
"resolved": "https://registry.npmjs.org/console-table-printer/-/console-table-printer-2.14.6.tgz",
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@
|
|||
"format": "biome check --write app components"
|
||||
},
|
||||
"dependencies": {
|
||||
"@ag-ui/client": "^0.0.40",
|
||||
"@ag-ui/client": "^0.0.42",
|
||||
"@copilotkit/react-core": "^1.10.6",
|
||||
"@copilotkit/react-ui": "^1.10.6",
|
||||
"@copilotkit/runtime": "^1.10.6",
|
||||
|
|
|
|||
Loading…
Reference in a new issue