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",