1.0 Release
This commit is contained in:
+2
-3
@@ -6,10 +6,9 @@ CFG = load_config()
|
||||
API_BASE = CFG["api"]["base_url"]
|
||||
API_VERSION = CFG["api"]["api_version"]
|
||||
|
||||
|
||||
class LocalLMEmbeddings(Embeddings):
|
||||
def __init__(
|
||||
self, model: str, base_url: str = API_BASE, batch_size: int = 32
|
||||
):
|
||||
def __init__(self, model: str, base_url: str = API_BASE, batch_size: int = 32):
|
||||
self.url = f"{base_url}/{API_VERSION}embeddings"
|
||||
self.model = model
|
||||
self.batch_size = batch_size
|
||||
|
||||
Reference in New Issue
Block a user