Fix: Type defs

zya · Oct 11, 2023 · ab5bfac · ab5bfac
1 parent 153f5f4
commit ab5bfac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -16,6 +16,8 @@ export async function completion(
   params: HandlerParamsStreaming,
 ): Promise<ResultStreaming>;
 
+export async function completion(params: HandlerParams): Promise<Result>;
+
 export async function completion(params: HandlerParams): Promise<Result> {
   const handler = getHandler(params.model);
 

diff --git a/tests/index.test.ts b/tests/index.test.ts
@@ -57,7 +57,7 @@ describe('litellm', () => {
       mockCreate.mockResolvedValueOnce({
         completion: 'response text',
       });
-      const result = (await completion(params)) as ResultNotStreaming;
+      const result = await completion(params);
       const expectedParams = {
         model: 'claude-2',
         prompt: expectedPrompt,