diff --git a/pgml-sdks/pgml/build.rs b/pgml-sdks/pgml/build.rs
index 82b51670c..f017a04db 100644
--- a/pgml-sdks/pgml/build.rs
+++ b/pgml-sdks/pgml/build.rs
@@ -8,6 +8,8 @@ async def migrate() -> None
 
 Json = Any
 DateTime = int
+GeneralJsonIterator = Any
+GeneralJsonAsyncIterator = Any
 "#;
 
 const ADDITIONAL_DEFAULTS_FOR_JAVASCRIPT: &[u8] = br#"
@@ -16,6 +18,8 @@ export function migrate(): Promise<void>;
 
 export type Json = any;
 export type DateTime = Date;
+export type GeneralJsonIterator = any;
+export type GeneralJsonAsyncIterator = any;
 
 export function newCollection(name: string, database_url?: string): Collection;
 export function newModel(name?: string, source?: string, parameters?: Json): Model;
diff --git a/pgml-sdks/pgml/python/tests/test.py b/pgml-sdks/pgml/python/tests/test.py
index f3b1fbec9..5c3a4df33 100644
--- a/pgml-sdks/pgml/python/tests/test.py
+++ b/pgml-sdks/pgml/python/tests/test.py
@@ -361,7 +361,6 @@ async def test_open_source_ai_create_async():
         ],
         temperature=0.85,
     )
-    import json
     assert len(results["choices"]) > 0
 
 
diff --git a/pgml-sdks/pgml/src/languages/javascript.rs b/pgml-sdks/pgml/src/languages/javascript.rs
index c9a09326d..c49b5c493 100644
--- a/pgml-sdks/pgml/src/languages/javascript.rs
+++ b/pgml-sdks/pgml/src/languages/javascript.rs
@@ -101,13 +101,13 @@ fn transform_stream_iterate_next(mut cx: FunctionContext) -> JsResult<JsPromise>
                         .expect("Error converting rust Json to JavaScript Object");
                     let d = cx.boolean(false);
                     o.set(&mut cx, "value", v)
-                        .expect("Error setting object value in transform_sream_iterate_next");
+                        .expect("Error setting object value in transform_stream_iterate_next");
                     o.set(&mut cx, "done", d)
-                        .expect("Error setting object value in transform_sream_iterate_next");
+                        .expect("Error setting object value in transform_stream_iterate_next");
                 } else {
                     let d = cx.boolean(true);
                     o.set(&mut cx, "done", d)
-                        .expect("Error setting object value in transform_sream_iterate_next");
+                        .expect("Error setting object value in transform_stream_iterate_next");
                 }
                 Ok(o)
             })