Update edgedb/ai alias package

geldata · Feb 22, 2025 · 09521b6 · 09521b6
1 parent f968138
commit 09521b6
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/edgedb/ai/index.ts b/edgedb/ai/index.ts
@@ -1,15 +1,24 @@
 import {
-  GelMessage,
-  GelSystemMessage,
-  GelUserMessage,
-  GelAssistantMessage,
-  GelToolMessage,
+  createRAGClient,
+  RAGClient,
+  Message,
+  SystemMessage,
+  UserMessage,
+  AssistantMessage,
+  ToolMessage,
+  RAGOptions,
 } from "@gel/ai";
 
 export * from "@gel/ai";
 
-export type EdgeDBSystemMessage = GelSystemMessage;
-export type EdgeDBUserMessage = GelUserMessage;
-export type EdgeDBAssistantMessage = GelAssistantMessage;
-export type EdgeDBToolMessage = GelToolMessage;
-export type EdgeDBMessage = GelMessage;
+export type EdgeDBAI = RAGClient;
+export const EdgeDBAI = RAGClient;
+
+export const createAI = createRAGClient;
+
+export type EdgeDBSystemMessage = SystemMessage;
+export type EdgeDBUserMessage = UserMessage;
+export type EdgeDBAssistantMessage = AssistantMessage;
+export type EdgeDBToolMessage = ToolMessage;
+export type EdgeDBMessage = Message;
+export type AIOptions = RAGOptions;