gered/llm-ls
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix: support raw TGI call (#10)

Browse source 
* fix: support raw TGI call

* feat: prepare `0.0.2` release
This commit is contained in:
Luc Georges 2023-09-07 19:11:05 +02:00 committed by GitHub
parent c774ec74fd
commit d8b7e05a20
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
crates/llm-ls/src/main.rs
View file

@ -84,6 +84,7 @@ impl Display for APIError {
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
#[serde(untagged)] #[serde(untagged)]
enum APIResponse { enum APIResponse {
Generation(Generation),
Generations(Vec<Generation>), Generations(Vec<Generation>),
Error(APIError), Error(APIError),
} }
@ -240,6 +241,7 @@ async fn request_completion(
let res = req.send().await.map_err(internal_error)?; let res = req.send().await.map_err(internal_error)?;
match res.json().await.map_err(internal_error)? { match res.json().await.map_err(internal_error)? {
APIResponse::Generation(gen) => Ok(vec![gen]),
APIResponse::Generations(gens) => Ok(gens), APIResponse::Generations(gens) => Ok(gens),
APIResponse::Error(err) => Err(internal_error(err)), APIResponse::Error(err) => Err(internal_error(err)),
} }
@ -374,7 +376,7 @@ impl LanguageServer for Backend {
Ok(InitializeResult { Ok(InitializeResult {
server_info: Some(ServerInfo { server_info: Some(ServerInfo {
name: "llm-ls".to_owned(), name: "llm-ls".to_owned(),
version: Some("0.1.1".to_owned()), version: Some("0.0.2".to_owned()),
}), }),
capabilities: ServerCapabilities { capabilities: ServerCapabilities {
text_document_sync: Some(TextDocumentSyncCapability::Kind( text_document_sync: Some(TextDocumentSyncCapability::Kind(