Fix for #803

- Fix related WASM issues
BoundaryML · Jul 19, 2024 · 5313401 · 5313401
1 parent 62085e7
commit 5313401
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 3 deletions.
diff --git a/engine/baml-runtime/src/lib.rs b/engine/baml-runtime/src/lib.rs
@@ -250,6 +250,7 @@ impl BamlRuntime {
             params,
             self.tracer.clone(),
             ctx.create_ctx(tb, cb)?,
+            #[cfg(not(target_arch = "wasm32"))]
             self.async_runtime.clone(),
         )
     }

diff --git a/engine/baml-runtime/src/runtime/runtime_interface.rs b/engine/baml-runtime/src/runtime/runtime_interface.rs
@@ -353,7 +353,7 @@ impl RuntimeInterface for InternalBamlRuntime {
         params: &BamlMap<String, BamlValue>,
         tracer: Arc<BamlTracer>,
         ctx: RuntimeContext,
-        tokio_runtime: Arc<tokio::runtime::Runtime>,
+        #[cfg(not(target_arch = "wasm32"))] tokio_runtime: Arc<tokio::runtime::Runtime>,
     ) -> Result<FunctionResultStream> {
         let func = self.get_function(&function_name, &ctx)?;
         let renderer = PromptRenderer::from_function(&func, self.ir(), &ctx)?;
@@ -373,6 +373,7 @@ impl RuntimeInterface for InternalBamlRuntime {
             orchestrator,
             tracer,
             renderer,
+            #[cfg(not(target_arch = "wasm32"))]
             tokio_runtime,
         })
     }

diff --git a/engine/baml-runtime/src/runtime_interface.rs b/engine/baml-runtime/src/runtime_interface.rs
@@ -43,7 +43,7 @@ pub trait RuntimeInterface {
         params: &BamlMap<String, BamlValue>,
         tracer: Arc<BamlTracer>,
         ctx: RuntimeContext,
-        tokio_runtime: Arc<tokio::runtime::Runtime>,
+        #[cfg(not(target_arch = "wasm32"))] tokio_runtime: Arc<tokio::runtime::Runtime>,
     ) -> Result<FunctionResultStream>;
 }
 

diff --git a/engine/baml-runtime/src/types/stream.rs b/engine/baml-runtime/src/types/stream.rs
@@ -27,6 +27,7 @@ pub struct FunctionResultStream {
     pub(crate) ir: Arc<IntermediateRepr>,
     pub(crate) orchestrator: OrchestratorNodeIterator,
     pub(crate) tracer: Arc<BamlTracer>,
+    #[cfg(not(target_arch = "wasm32"))]
     pub(crate) tokio_runtime: Arc<tokio::runtime::Runtime>,
 }
 
@@ -54,6 +55,7 @@ first.scope.clone();
 */
 
 impl FunctionResultStream {
+    #[cfg(not(target_arch = "wasm32"))]
     pub fn run_sync<F>(
         &mut self,
         on_event: Option<F>,

diff --git a/typescript/vscode-ext/packages/language-server/src/server.ts b/typescript/vscode-ext/packages/language-server/src/server.ts
@@ -33,7 +33,7 @@ import fs from 'fs'
 // import { cliBuild, cliCheckForUpdates, cliVersion } from './baml-cli'
 import { type ParserDatabase, TestRequest } from '@baml/common'
 import debounce from 'lodash/debounce'
-import { TextDocumentIdentifier } from 'vscode-languageserver-protocol'
+// import { TextDocumentIdentifier } from 'vscode-languageserver-protocol'
 import { TextDocument } from 'vscode-languageserver-textdocument'
 import { IPCMessageReader, IPCMessageWriter, createConnection } from 'vscode-languageserver/node'
 import { z } from 'zod'