@letta-ai/letta-react 0.2.0 → 0.2.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@letta-ai/letta-react",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Letta's react library",
   "private": false,
   "main": "./index.js",

package/src/index.ts

@@ -1,2 +1 @@
 export * from './useLettaQuery/useLettaQuery';
-export * from './useLetta/useLetta';

package/src/useLettaQuery/useLettaQuery.test.tsx

@@ -1,12 +1,10 @@
 import React from 'react';
 import { renderHook, waitFor } from '@testing-library/react';
-import { useLettaQuery } from './useLettaQuery.ts';
-import { LettaClient } from '@letta-ai/letta-client';
+import { useLettaQuery } from './useLettaQuery';
 import { enableFetchMocks } from 'jest-fetch-mock';
 import fetchMock from 'jest-fetch-mock';
 enableFetchMocks();
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
-import { useLetta } from '../useLetta/useLetta.ts';
 
 const queryClient = new QueryClient({
   defaultOptions: {
@@ -34,9 +32,14 @@ describe('useLettaQuery', () => {
 
     const { result } = renderHook(
       () => {
-        const client = useLetta();
-
-        return useLettaQuery((args) => client.agents.list(args), { limit: 1 });
+        return useLettaQuery(
+          (client) => {
+            return client.agents.list();
+          },
+          {
+            queryKey: ['agents'],
+          }
+        );
       },
       { wrapper }
     );

package/src/useLettaQuery/useLettaQuery.ts

@@ -1,34 +1,22 @@
 import { useQuery } from '@tanstack/react-query';
+import { useRef } from 'react';
+import { LettaClient } from '@letta-ai/letta-client';
+import type { UseQueryOptions } from '@tanstack/react-query/src/types.ts';
 
-export function getQueryKey<Request>(
-  operationName: string,
-  request: Request
-): [string, Request] {
-  return [operationName, request];
-}
+type Query<Res> = (client: LettaClient) => Res;
+
+type Options<Res> = Omit<UseQueryOptions<Res>, 'queryFn'>;
+
+export function useLettaQuery<Res>(
+  operation: Query<Promise<Res>>,
+  queryOptions: Options<Res>
+) {
+  const client = useRef(new LettaClient());
 
-export function useLettaQuery<
-  Request extends Record<string, any> = Record<string, any>,
-  Options extends Record<string, any> = Record<string, any>,
-  Response = unknown
->(
-  operation: (request: Request, options?: Options) => Response,
-  request: Request,
-  options?: Options,
-  queryOptions = {}
-): ReturnType<
-  typeof useQuery<
-    Awaited<Response>,
-    Error,
-    Awaited<Response>,
-    [string, Request]
-  >
-> {
-  return useQuery<Response, Error, Awaited<Response>, [string, Request]>({
-    queryKey: getQueryKey<Request>(operation.name, request),
-    queryFn: () => {
-      return operation(request, options);
+  return useQuery<Res>({
+    ...queryOptions,
+    queryFn: async () => {
+      return operation(client.current);
     },
-    ...(queryOptions || {}),
   });
 }

package/useLettaQuery/useLettaQuery.d.ts

@@ -1,6 +1,6 @@
-import { useQuery } from '@tanstack/react-query';
-export declare function getQueryKey<Request>(operationName: string, request: Request): [string, Request];
-export declare function useLettaQuery<Request extends Record<string, any> = Record<string, any>, Options extends Record<string, any> = Record<string, any>, Response = unknown>(operation: (request: Request, options?: Options) => Response, request: Request, options?: Options, queryOptions?: {}): ReturnType<typeof useQuery<Awaited<Response>, Error, Awaited<Response>, [
-    string,
-    Request
-]>>;
+import { LettaClient } from '@letta-ai/letta-client';
+import type { UseQueryOptions } from '@tanstack/react-query/src/types.ts';
+type Query<Res> = (client: LettaClient) => Res;
+type Options<Res> = Omit<UseQueryOptions<Res>, 'queryFn'>;
+export declare function useLettaQuery<Res>(operation: Query<Promise<Res>>, queryOptions: Options<Res>): import("@tanstack/react-query").UseQueryResult<Res, Error>;
+export {};

package/dist/useLetta/useLetta.d.ts

@@ -1,2 +0,0 @@
-import { LettaClient } from '@letta-ai/letta-client';
-export declare function useLetta(options?: LettaClient.Options): LettaClient;

package/src/useLetta/useLetta.ts

@@ -1,8 +0,0 @@
-import { useState } from 'react';
-import { LettaClient } from '@letta-ai/letta-client';
-
-export function useLetta(options: LettaClient.Options = {}) {
-  const [client] = useState(new LettaClient(options));
-
-  return client;
-}

package/useLetta/useLetta.d.ts

@@ -1,2 +0,0 @@
-import { LettaClient } from '@letta-ai/letta-client';
-export declare function useLetta(options?: LettaClient.Options): LettaClient;