Skip to content

Commit 6d60290

Browse files
committed
WIP: Focus on remote
Signed-off-by: Aminu Oluwaseun Joshua <[email protected]>
1 parent 71dab62 commit 6d60290

File tree

4 files changed

+2
-34
lines changed

4 files changed

+2
-34
lines changed

crates/llm-local/src/lib.rs

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
mod bert;
22
mod llama;
3-
mod open_ai;
43

54
use anyhow::Context;
65
use bert::{BertModel, Config};
@@ -30,7 +29,6 @@ pub struct LocalLlmEngine {
3029
#[derive(Debug)]
3130
enum InferencingModelArch {
3231
Llama,
33-
_OpenAI,
3432
}
3533

3634
impl FromStr for InferencingModelArch {
@@ -156,11 +154,6 @@ impl LocalLlmEngine {
156154
.await
157155
.map_err(|e| wasi_llm::Error::RuntimeError(e.to_string()))?,
158156
),
159-
InferencingModelArch::_OpenAI => Arc::new(
160-
open_ai::OpenAIModels::new(&model_dir)
161-
.await
162-
.map_err(|e| wasi_llm::Error::RuntimeError(e.to_string()))?,
163-
),
164157
};
165158

166159
v.insert(model.clone());

crates/llm-local/src/open_ai.rs

Lines changed: 0 additions & 27 deletions
This file was deleted.

crates/llm-remote-http/src/lib.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ use serde::{Deserialize, Serialize};
77
use serde_json::json;
88
use spin_world::v2::llm::{self as wasi_llm};
99

10+
mod open_ai;
11+
1012
#[derive(Clone)]
1113
pub struct RemoteHttpLlmEngine {
1214
auth_token: String,

crates/llm-remote-http/src/open_ai.rs

Whitespace-only changes.

0 commit comments

Comments
 (0)