General clean ups (#1208)

SilasMarvin · web-flow · commit c8d595836a8f · 2023-12-01T13:40:38.000-08:00
diff --git 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
@@ -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
@@ -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)
             })

Original file line number	Diff line number	Diff line change
`@@ -361,7 +361,6 @@ async def test_open_source_ai_create_async():`
`361`	`361`	`],`
`362`	`362`	`temperature=0.85,`
`363`	`363`	`)`
`364`		`- import json`
`365`	`364`	`assert len(results["choices"]) > 0`
`366`	`365`
`367`	`366`