From 7dff30f9cbcfa3654d08a8559f5fe9ed097f7f46 Mon Sep 17 00:00:00 2001 From: "Hauzer S. Lee" Date: Wed, 16 Oct 2024 15:19:41 +0800 Subject: [PATCH] chore: files added --- llmsearch/clients/__init__.py | 0 llmsearch/clients/localai.py | 2 +- requirements.txt | 2 ++ 3 files changed, 3 insertions(+), 1 deletion(-) create mode 100644 llmsearch/clients/__init__.py diff --git a/llmsearch/clients/__init__.py b/llmsearch/clients/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/llmsearch/clients/localai.py b/llmsearch/clients/localai.py index 3bd7d96..f6f8f8a 100644 --- a/llmsearch/clients/localai.py +++ b/llmsearch/clients/localai.py @@ -136,7 +136,7 @@ def embedding(self, model: str, text: str) -> Optional[List[float]]: # Example usage: if __name__ == "__main__": # Initialize the client - client = LocalAIClient(base_url="http://localhost:8080", model="your-model-name") + client = LocalAIClient(base_url="http://localhost:8080") # Example 1: Generating text prompt = "Tell me a story about a brave knight." diff --git a/requirements.txt b/requirements.txt index 479e528..0636b3d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,3 +14,5 @@ alembic==1.13.3 black==24.10.0 Flask-SQLAlchemy==3.1.1 bcrypt==4.2.0 +tensorflow==2.17.0 +transformers==4.45.2