From a539392ad48f9d03696587cb49ac595782a6f95f Mon Sep 17 00:00:00 2001
From: lyblsgo <lyblsgo@163.com>
Date: 星期四, 27 四月 2023 15:17:24 +0800
Subject: [PATCH] fix paraformer server for new apis
---
funasr/runtime/grpc/paraformer-server.h | 9 +++++++--
1 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/funasr/runtime/grpc/paraformer_server.h b/funasr/runtime/grpc/paraformer-server.h
similarity index 70%
rename from funasr/runtime/grpc/paraformer_server.h
rename to funasr/runtime/grpc/paraformer-server.h
index dba1e45..108e3b6 100644
--- a/funasr/runtime/grpc/paraformer_server.h
+++ b/funasr/runtime/grpc/paraformer-server.h
@@ -37,13 +37,18 @@
float snippet_time;
}FUNASR_RECOG_RESULT;
-
class ASRServicer final : public ASR::Service {
private:
int init_flag;
+ std::unordered_map<std::string, std::string> client_buffers;
+ std::unordered_map<std::string, std::string> client_transcription;
public:
- ASRServicer(const char* model_path, int thread_num, bool quantize);
+ ASRServicer(std::map<std::string, std::string>& model_path);
+ void clear_states(const std::string& user);
+ void clear_buffers(const std::string& user);
+ void clear_transcriptions(const std::string& user);
+ void disconnect(const std::string& user);
grpc::Status Recognize(grpc::ServerContext* context, grpc::ServerReaderWriter<Response, Request>* stream);
FUNASR_HANDLE AsrHanlde;
--
Gitblit v1.9.1