Skip to content
This repository was archived by the owner on Jun 23, 2026. It is now read-only.

Added row embedding support #126

Merged
cybermaggedon merged 1 commit into from
Feb 23, 2026
Merged

Added row embedding support #126

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 15 additions & 15 deletions package-lock.json
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions package.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,9 @@
"@msgpack/msgpack": "^3.1.1",
"@tanstack/react-query": "^5.80.3",
"@tanstack/react-table": "^8.21.3",
"@trustgraph/client": "^1.4.0",
"@trustgraph/react-provider": "^1.3.0",
"@trustgraph/react-state": "^1.4.0",
"@trustgraph/client": "^1.4.2",
"@trustgraph/react-provider": "^1.4.0",
"@trustgraph/react-state": "^1.4.1",
"@types/dagre": "^0.7.53",
"compute-cosine-similarity": "^1.1.0",
"dagre": "^0.8.5",
Expand Down
48 changes: 46 additions & 2 deletions src/components/agents/EditDialog.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import { usePrompts } from "@trustgraph/react-state";
import SelectField from "../common/SelectField";
import TextAreaField from "../common/TextAreaField";
import TextField from "../common/TextField";
import NumberField from "../common/NumberField";
import ChipInputField from "../common/ChipInputField";
import { toaster } from "../ui/toaster";
import EditableArgumentsTable from "./EditableArgumentsTable";
Expand All @@ -29,6 +30,9 @@ const EditDialog = ({ open, onOpenChange, onComplete, id, create }) => {
const [templateId, setTemplateId] = useState("");
const [mcpToolId, setMcpToolId] = useState("");
const [collection, setCollection] = useState("");
const [schemaName, setSchemaName] = useState("");
const [indexName, setIndexName] = useState("");
const [limit, setLimit] = useState(10);
const [group, setGroup] = useState([]);
const [state, setState] = useState("");
const [applicableStates, setApplicableStates] = useState([]);
Expand Down Expand Up @@ -56,6 +60,10 @@ const EditDialog = ({ open, onOpenChange, onComplete, id, create }) => {
setMcpToolId(x.mcp_tool_id || x["mcp-tool"] || "");
// Handle collection attribute for knowledge-query tools
setCollection(x.collection || "");
// Handle row-embeddings-query fields
setSchemaName(x["schema-name"] || "");
setIndexName(x["index-name"] || "");
setLimit(x.limit || 10);
// Handle new optional fields
setGroup(x.group || []);
setState(x.state || "");
Expand Down Expand Up @@ -87,6 +95,12 @@ const EditDialog = ({ open, onOpenChange, onComplete, id, create }) => {
description:
"Execute natural language questions against records in a structured data / object store",
},
{
value: "row-embeddings-query",
label: "Row Embeddings Query",
description:
"Semantic search on structured data indexes using vector similarity",
},
{
value: "mcp-tool",
label: "MCP Tool",
Expand Down Expand Up @@ -125,8 +139,12 @@ const EditDialog = ({ open, onOpenChange, onComplete, id, create }) => {
arguments: args,
...(type === "prompt" && templateId && { template: templateId }),
...(type === "mcp-tool" && mcpToolId && { "mcp-tool": mcpToolId }),
...((type === "knowledge-query" || type === "structured-query") &&
...((type === "knowledge-query" || type === "structured-query" || type === "row-embeddings-query") &&
collection && { collection: collection }),
// Row embeddings query specific fields
...(type === "row-embeddings-query" && schemaName && { "schema-name": schemaName }),
...(type === "row-embeddings-query" && indexName && { "index-name": indexName }),
...(type === "row-embeddings-query" && { limit: limit }),
...(group && group.length > 0 && { group: group }),
...(state && { state: state }),
...(applicableStates &&
Expand Down Expand Up @@ -253,7 +271,7 @@ const EditDialog = ({ open, onOpenChange, onComplete, id, create }) => {
/>
)}

{(type === "knowledge-query" || type === "structured-query") && (
{(type === "knowledge-query" || type === "structured-query" || type === "row-embeddings-query") && (
<TextField
label="Collection"
placeholder="Enter the knowledge collection (optional)"
Expand All @@ -263,6 +281,32 @@ const EditDialog = ({ open, onOpenChange, onComplete, id, create }) => {
/>
)}

{type === "row-embeddings-query" && (
<>
<TextField
label="Schema Name"
placeholder="Enter the schema name for the structured data"
value={schemaName}
onValueChange={(v) => setSchemaName(v)}
required={true}
/>
<TextField
label="Index Name"
placeholder="Enter specific index name to search (optional)"
value={indexName}
onValueChange={(v) => setIndexName(v)}
required={false}
/>
<NumberField
label="Limit"
value={limit}
onValueChange={(v) => setLimit(v)}
minValue={1}
maxValue={100}
/>
</>
)}

<ChipInputField
label="Groups"
values={group}
Expand Down
Loading