Export more types

arshad-yaseen · Sep 15, 2024 · c14ed35 · c14ed35
1 parent 53bc88a
commit c14ed35
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -9,6 +9,9 @@ export type {
   RegisterCopilotOptions,
   CopilotRegistration,
   CopilotOptions,
+  CustomModel,
+  CustomModelConfig,
+  CustomModelTransformResponse,
   CompletionMetadata,
   CompletionRequest,
   CompletionRequestBody,

diff --git a/src/types/copilot.ts b/src/types/copilot.ts
@@ -58,7 +58,7 @@ export type CustomModel = {
    *          text to be inserted or used as the completion, without
    *          any metadata or additional structure.
    */
-  transformResponse: CustomModelResponse;
+  transformResponse: CustomModelTransformResponse;
 };
 
 export type CustomModelConfig = (
@@ -85,7 +85,9 @@ export type CustomModelConfig = (
   body?: Record<string, unknown>;
 };
 
-export type CustomModelResponse = (response: unknown) => CompletionResponse;
+export type CustomModelTransformResponse = (
+  response: unknown,
+) => CompletionResponse;
 
 export type Endpoint = string;
 export type Filename = string;

diff --git a/tests/provider-handlers.test.ts b/tests/provider-handlers.test.ts
@@ -1,11 +1,7 @@
 import {describe, expect, it} from 'vitest';
 
 import {parseProviderChatCompletion} from '../src/helpers/provider';
-import {
-  ChatCompletion,
-  CompletionProvider,
-  PickChatCompletion,
-} from '../src/types';
+import {ChatCompletion, PickChatCompletion} from '../src/types';
 
 describe('Provider Handler Functions', () => {
   describe('parseProviderChatCompletion', () => {