Merge pull request #23 from AI21Labs/add_types_to_tsconfig

fix: Added types to index

Author: Josephasafg

package.json

@@ -8,6 +8,7 @@
   "type": "module",
   "exports": {
     ".": {
+      "types": "./dist/index.d.ts",
       "require": "./dist/bundle.cjs.js",
       "import": "./dist/bundle.es.js"
     }

src/index.ts

@@ -1,7 +1,46 @@
 export { AI21 } from './AI21';
 export { VERSION } from './version';
-export * from './types';
+export {
+  type DocumentSchema,
+  type ChatCompletionCreateParams,
+  type ChatCompletionCreateParamsNonStreaming,
+  type ChatCompletionCreateParamsStreaming,
+  type ChatMessage,
+  type AssistantMessage,
+  type ToolMessage,
+  type UserMessage,
+  type SystemMessage,
+  type ChatMessageParam,
+  type ToolDefinition,
+  type ResponseFormat,
+  type FunctionToolDefinition,
+  type ToolFunction,
+  type ToolParameters,
+  type ToolCall,
+  type ChatCompletionResponseChoice,
+  type ChatCompletionResponse,
+  type TopTokenData,
+  type LogprobsData,
+  type Logprobs,
+  type ChoiceDelta,
+  type ChoicesChunk,
+  type ChatCompletionChunk,
+  type ChatModel,
+  type UsageInfo,
+  type APIResponseProps,
+  type RequestOptions,
+  type FinalRequestOptions,
+  type HTTPMethod,
+  type DefaultQuery,
+  type Headers,
+  type CrossPlatformResponse,
+  type ConversationalRagRequest,
+  type ConversationalRagResponse,
+  type ConversationalRagSource,
+  type RetrievalStrategy,
+} from './types';
+export { APIClient } from './APIClient';
 export { AI21Error, MissingAPIKeyError } from './errors';
 export { Stream } from './streaming';
 export { APIResource } from './APIResource';
-export { Chat, Completions } from './resources';
+export { Chat, Completions, ConversationalRag } from './resources';