apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-06 03:42:11 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(ts): complete schema-first phase 2

Browse source
This commit is contained in:
elpresidank 2026-06-11 07:37:59 -05:00
parent 0746d7ffd5
commit be2370ee7b
24 changed files with 465 additions and 433 deletions
Download patch file Download diff file Expand all files Collapse all files

27
ts/packages/flow/src/retrieval/graph-rag.ts
View file

@ -16,23 +16,22 @@ import type {
Term,
TextCompletionRequest,
TextCompletionResponse,
Triple,
TriplesQueryRequest,
TriplesQueryResponse,
} from "@trustgraph/base";
import { errorMessage } from "@trustgraph/base";
import { Triple, errorMessage } from "@trustgraph/base";
import { Context, Effect, Layer, Match } from "effect";
import * as O from "effect/Option";
import * as S from "effect/Schema";
export interface GraphRagConfig {
entityLimit?: number;
tripleLimit?: number;
maxSubgraphSize?: number;
maxPathLength?: number;
edgeScoreLimit?: number;
edgeLimit?: number;
}
export class GraphRagConfig extends S.Class<GraphRagConfig>("GraphRagConfig")({
entityLimit: S.optionalKey(S.Finite),
tripleLimit: S.optionalKey(S.Finite),
maxSubgraphSize: S.optionalKey(S.Finite),
maxPathLength: S.optionalKey(S.Finite),
edgeScoreLimit: S.optionalKey(S.Finite),
edgeLimit: S.optionalKey(S.Finite),
}, { description: "Graph RAG retrieval tuning limits." }) {}
export interface GraphRagClients {
llm: EffectRequestResponse<TextCompletionRequest, TextCompletionResponse>;
@ -53,10 +52,10 @@ export interface GraphRagQueryOptions {
readonly chunkCallback?: ChunkCallback;
}
export interface GraphRagResult {
answer: string;
subgraph: Triple[];
}
export class GraphRagResult extends S.Class<GraphRagResult>("GraphRagResult")({
answer: S.String,
subgraph: S.Array(Triple),
}, { description: "Graph RAG answer with the supporting subgraph." }) {}
interface NormalizedGraphRagConfig {
entityLimit: number;