From 0587da73fea440a3127b1a58b4036ee7940e01bc Mon Sep 17 00:00:00 2001 From: Yeqi Huang Date: Mon, 18 Nov 2024 00:37:45 +0000 Subject: [PATCH] refactor: version 0.9.0, use "model + utils + core" to organize the project (#37) * Bump version 0.8.0 -> 0.8.1 * refactor: update module structure * Bump version 0.8.1 -> 0.9.0 * refactor: update all examples and init files * chore: format code * feat: support vllm & sllm inference (#34) * merge: merge vllm/sllm support * fix: fix module import in sllm/vllm --- nerif/utils/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/nerif/utils/__init__.py b/nerif/utils/__init__.py index 985adf1..d4b954b 100644 --- a/nerif/utils/__init__.py +++ b/nerif/utils/__init__.py @@ -26,6 +26,8 @@ get_litellm_embedding, get_litellm_response, get_ollama_response, + get_sllm_response, + get_vllm_response, similarity_dist, ) @@ -53,6 +55,8 @@ "get_litellm_embedding", "get_litellm_response", "get_ollama_response", + "get_sllm_response", + "get_vllm_response", # environment variables "OPENAI_API_BASE", "OPENAI_API_KEY",