@1kbirds/chidori 0.1.16

package/src/translations/rust.rs

@@ -0,0 +1,177 @@
+
+use anyhow::Result;
+
+use prompt_graph_core::graph_definition::DefinitionGraph;
+use prompt_graph_core::build_runtime_graph::graph_parse::CleanedDefinitionGraph;
+use prompt_graph_core::proto2::execution_runtime_client::ExecutionRuntimeClient;
+use prompt_graph_core::proto2::RequestFileMerge;
+
+
+// TODO: need to define pattern for automatically wiring dependencies
+// TODO: add methods
+// create_entrypoint_query
+
+pub fn new_graph() -> DefinitionGraph {
+    DefinitionGraph::zero()
+}
+
+#[macro_export]
+macro_rules! register_all {
+    ($graph:expr, $($statement:expr,)*) => {
+        $( $graph.register_node($statement); )*
+    };
+}
+
+#[macro_export]
+macro_rules! parameter {
+    ($output_def:expr) => {
+        create_node_parameter(
+            format!("{}:{}", file!(), line!()),
+            String::from($output_def)
+        )
+    };
+    ($name:expr, $output_def:expr) => {
+        create_node_parameter(
+            String::from($name),
+            String::from($output_def)
+        )
+    };
+}
+
+#[macro_export]
+macro_rules! map {
+    ($query_def:expr, $path:expr) => {
+        create_op_map(
+            format!("{}:{}", file!(), line!()),
+            String::from($query_def),
+            String::from($path),
+        )
+    };
+    ($name:expr, $query_def:expr, $path:expr) => {
+        create_code_node(
+            String::from($name),
+            String::from($query_def),
+            String::from($path),
+        )
+    };
+}
+
+
+#[macro_export]
+macro_rules! code_node {
+    ($query_def:expr, $output_def:expr, $source_type:expr) => {
+        create_code_node(
+            format!("{}:{}", file!(), line!()),
+            String::from($query_def),
+            String::from($output_def),
+            $source_type,
+        )
+    };
+    ($name:expr, $query_def:expr, $output_def:expr, $source_type:expr) => {
+        create_code_node(
+            String::from($name),
+            String::from($query_def),
+            String::from($output_def),
+            String::from($source_type),
+        )
+    };
+}
+
+
+#[macro_export]
+macro_rules! observation_node {
+    ($query_def:expr, $output_def:expr) => {
+        create_observation_node(
+            format!("{}:{}", file!(), line!()),
+            String::from($query_def),
+            String::from($output_def),
+        )
+    };
+    ($name:expr, $query_def:expr, $output_def:expr) => {
+        create_observation_node(
+            String::from($name),
+            String::from($query_def),
+            String::from($output_def),
+        )
+    };
+}
+
+
+#[macro_export]
+macro_rules! memory_node {
+    ($query_def:expr, $output_def:expr, $model:expr, $db:expr) => {
+        create_vector_memory_node(
+            format!("{}:{}", file!(), line!()),
+            String::from($query_def),
+            String::from($output_def),
+            String::from($model),
+            String::from($db)
+        )
+    };
+    ($name:expr, $query_def:expr, $output_def:expr, $model:expr, $db:expr) => {
+        create_vector_memory_node(
+            String::from($name),
+            String::from($query_def),
+            String::from($output_def),
+            String::from($model),
+            String::from($db)
+        )
+    };
+}
+
+#[macro_export]
+macro_rules! component_node {
+    ($query_def:expr, $output_def:expr) => {
+        create_component_node(
+            format!("{}:{}", file!(), line!()),
+            String::from($query_def),
+            String::from($output_def),
+        )
+    };
+    ($name:expr, $query_def:expr, $output_def:expr) => {
+        create_component_node(
+            String::from($name),
+            String::from($query_def),
+            String::from($output_def),
+        )
+    };
+}
+
+#[macro_export]
+macro_rules! prompt_node {
+    ($query_def:expr, $template:expr, $model:expr) => {
+        create_prompt_node(
+            format!("{}:{}", file!(), line!()),
+            String::from($query_def),
+            String::from($template),
+            String::from($model),
+        )
+    };
+    ($name:expr, $query_def:expr, $template:expr, $model:expr) => {
+        create_prompt_node(
+            String::from($name),
+            String::from($query_def),
+            String::from($template),
+            String::from($model),
+        )
+    };
+}
+
+pub async fn run_worker(url: String, graph: DefinitionGraph) -> Result<()> {
+    // validate the graph
+    CleanedDefinitionGraph::new(&graph);
+    let file_merge = RequestFileMerge{ file: Some(graph.get_file().clone()), branch: 0, id: "".to_string() };
+    let mut client = ExecutionRuntimeClient::connect(url).await?;
+    let _response = client.merge(file_merge).await?.into_inner();
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::new_graph;
+
+    #[test]
+    fn test_new_graph() {
+        new_graph();
+    }
+}

package/src/translations/wasm.rs

@@ -0,0 +1 @@
+// https://crates.io/crates/grpc-web-client

package/tests/nodejs/chidori.test.js

@@ -0,0 +1,54 @@
+const {Chidori} = require("../..");
+
+async function delay(ms) {
+    // Returns a promise that resolves after "ms" milliseconds
+    return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+
+test('initialize without error', () => {
+    expect(new Chidori("1", "localhost:9800")).toEqual({"chi": {}});
+});
+
+test('coerce to and from structure', async () => {
+    const chi = new Chidori("1", "http://localhost:9800");
+    chi.startServer()
+    expect(chi.objectInterface({
+        "id": "1",
+        "monotonic_counter": 0,
+        "branch": 0
+    })).toEqual({
+        "id": "1",
+        "monotonic_counter": 0,
+        "branch": 0
+    });
+});
+
+
+test('start server', async () => {
+    const chi = new Chidori("21", "http://127.0.0.1:9800");
+    await chi.startServer();
+    await chi.play(0,0);
+    await chi.denoCodeNode({
+        name: "InspirationalQuote",
+        code: `return {"promptResult": "Believe"}`,
+        output: `type InspirationalQuote { promptResult: String }`,
+        is_template: true
+    });
+    await chi.denoCodeNode({
+        name: "CodeNode",
+        queries: ["query Q { InspirationalQuote { promptResult } }"],
+        code: `return {"output": "Here is your quote for " + \`{{InspirationalQuote.promptResult}}\` }`,
+        output: `type CodeNode { output: String }`,
+        is_template: true
+    });
+    await delay(1000);
+    console.log(await chi.graphStructure())
+    expect((await chi.query(`
+        query Q { InspirationalQuote { promptResult } }
+    `, 0, 100))["values"].length).toBe(1)
+    console.log(await chi.query(`
+        query Q { CodeNode { output } }
+    `, 0, 100))
+});
+

package/tests/nodejs/nodeHandle.test.js

@@ -0,0 +1,17 @@
+const {simpleFun, NodeHandle} = require("../..");
+const {Chidori} = require("../../package_node");
+
+test('sdk', () => {
+    expect(simpleFun("1")).toBe("1");
+});
+
+test('adds 1 + 2 to equal 3', () => {
+    expect(new NodeHandle()).toEqual({"nh": {}});
+});
+
+test('nodehandle query', async () => {
+    const chi = new Chidori("1", "http://localhost:9800");
+    await chi.startServer();
+    const nh = new NodeHandle();
+    expect(await nh.query(0, 0)).toEqual({});
+});