Merge pull request #7 from parea-ai/export-new-types

fix(exports): export new types from index
parea-ai · Jan 29, 2024 · c0770ac · c0770ac
2 parents 7d3b202 + f09d829
commit c0770ac
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/experiment/experiment.ts b/src/experiment/experiment.ts
@@ -44,7 +44,7 @@ function calculateAvgStdForExperiment(experimentStats: ExperimentStatsSchema): {
   return result;
 }
 
-export async function experiment(
+async function experiment(
   name: string,
   data: Iterable<Record<string, any>>,
   func: (...args: any[]) => Promise<any>,

diff --git a/src/index.ts b/src/index.ts
@@ -7,7 +7,9 @@ export { Parea } from './client';
 
 export { pareaLogger } from './parea_logger';
 
-export { genTraceId, toDateTimeString } from './helpers';
+export { genTraceId, toDateTimeString, asyncPool } from './helpers';
+
+export { Experiment } from './experiment/experiment';
 
 export {
   Role,
@@ -21,6 +23,16 @@ export {
   UseDeployedPromptResponse,
   FeedbackRequest,
   TraceLogInputs,
+  Log,
   TraceLog,
   TraceLogTreeSchema,
+  NamedEvaluationScore,
+  TraceOptions,
+  UpdateLog,
+  CreateExperimentRequest,
+  ExperimentSchema,
+  EvaluationScoreSchema,
+  TraceStatsSchema,
+  ExperimentStatsSchema,
+  DataItem,
 } from './types';