Merge pull request #33 from evaline-ju/basic-ci

👷 Add initial CI

Author: evaline-ju

.github/workflows/build.yml

@@ -0,0 +1,26 @@
+name: Build Orchestrator
+
+on:
+  push:
+    branches: ["main"]
+  pull_request:
+    branches: ["main"]
+
+env:
+  CARGO_TERM_COLOR: always
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+      - name: Build
+        run: cargo build
+      - name: Lint with clippy
+        run: cargo clippy --all-targets --all-features -- -D warnings
+      - name: Check formatting
+        run: cargo fmt -- -check
+      - name: Test
+        run: cargo test
+     

src/clients/chunker.rs

@@ -14,9 +14,7 @@ use crate::{
             chunkers_service_client::ChunkersServiceClient, BidiStreamingTokenizationTaskRequest,
             TokenizationTaskRequest,
         },
-        caikit_data_model::nlp::{
-            TokenizationResults, TokenizationStreamResult,
-        },
+        caikit_data_model::nlp::{TokenizationResults, TokenizationStreamResult},
     },
 };
 
@@ -73,7 +71,6 @@ impl ChunkerClient {
         });
         Ok(ReceiverStream::new(rx))
     }
-
 }
 
 fn request_with_model_id<T>(request: T, model_id: &str) -> Request<T> {

src/clients/nlp.rs

@@ -1,6 +1,6 @@
-use std::{collections::HashMap, pin::Pin};
+use std::collections::HashMap;
 
-use futures::{Stream, StreamExt};
+use futures::StreamExt;
 use ginepro::LoadBalancedChannel;
 use tokio::sync::mpsc;
 use tokio_stream::wrappers::ReceiverStream;
@@ -11,9 +11,8 @@ use crate::{
     config::ServiceConfig,
     pb::{
         caikit::runtime::nlp::{
-            nlp_service_client::NlpServiceClient,
-            ServerStreamingTextGenerationTaskRequest, TextGenerationTaskRequest,
-            TokenClassificationTaskRequest, TokenizationTaskRequest,
+            nlp_service_client::NlpServiceClient, ServerStreamingTextGenerationTaskRequest,
+            TextGenerationTaskRequest, TokenClassificationTaskRequest, TokenizationTaskRequest,
         },
         caikit_data_model::nlp::{
             GeneratedTextResult, GeneratedTextStreamResult, TokenClassificationResults,

src/orchestrator.rs

@@ -10,7 +10,8 @@ use uuid::Uuid;
 
 use crate::{
     clients::{
-        self, detector::DetectorRequest, ChunkerClient, DetectorClient, GenerationClient, NlpClient, TgisClient,
+        self, detector::DetectorRequest, ChunkerClient, DetectorClient, GenerationClient,
+        NlpClient, TgisClient,
     },
     config::{GenerationProvider, OrchestratorConfig},
     models::{