Merge branch 'main' of github.com:Chainlit/typescript-client into wil…

…ly/eng-1669-experiments-features

src/index.ts

@@ -75,6 +75,18 @@ export class LiteralClient {
     });
   }
 
+  _currentThread(): Thread | null {
+    const store = storage.getStore();
+
+    return store?.currentThread || null;
+  }
+
+  _currentStep(): Step | null {
+    const store = storage.getStore();
+
+    return store?.currentStep || null;
+  }
+
   /**
    * Gets the current thread from the context.
    * WARNING : this will throw if run outside of a thread context.

tests/attachments.test.ts

@@ -3,12 +3,12 @@ import { createReadStream, readFileSync } from 'fs';
 
 import { Attachment, LiteralClient, Maybe } from '../src';
 
-const url = process.env.LITERAL_API_URL;
+const apiUrl = process.env.LITERAL_API_URL;
 const apiKey = process.env.LITERAL_API_KEY;
-if (!url || !apiKey) {
+if (!apiUrl || !apiKey) {
   throw new Error('Missing environment variables');
 }
-const client = new LiteralClient(apiKey, url);
+const client = new LiteralClient({ apiKey, apiUrl });
 
 const filePath = './tests/chainlit-logo.png';
 const mime = 'image/png';

tests/integration/openai.test.ts

@@ -348,7 +348,7 @@ describe('OpenAI Instrumentation', () => {
 
   describe('Handling tags and metadata', () => {
     it('handles tags and metadata on the instrumentation call', async () => {
-      const client = new LiteralClient(apiKey, url);
+      const client = new LiteralClient({ apiKey, apiUrl });
       client.instrumentation.openai({
         tags: ['tag1', 'tag2'],
         metadata: { key: 'value' }
@@ -382,7 +382,7 @@ describe('OpenAI Instrumentation', () => {
     }, 30_000);
 
     it('handles tags and metadata on the LLM call', async () => {
-      const client = new LiteralClient(apiKey, url);
+      const client = new LiteralClient({ apiKey, apiUrl });
 
       const instrumentedOpenAi = client.instrumentation.openai({
         tags: ['tag1', 'tag2'],