Skip to content

Commit 8afd19d

Browse files
kkopczynski-cyborgdupontcyborg
authored andcommitted
removed https requirement from client
1 parent 086e22d commit 8afd19d

File tree

6 files changed

+7
-20
lines changed

6 files changed

+7
-20
lines changed

cyborgdb/client/client.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -86,14 +86,9 @@ class Client:
8686
"""
8787

8888
def __init__(self, api_url, api_key, verify_ssl=None):
89-
# Ensure the API URL uses HTTPS
89+
# If api_url is http, disable SSL verification
9090
if api_url.startswith('http://'):
91-
api_url = api_url.replace('http://', 'https://')
92-
logger.warning(f"Automatically converted HTTP URL to HTTPS: {api_url}")
93-
94-
# Validate that the URL uses HTTPS
95-
if not api_url.startswith('https://'):
96-
raise ValueError("API URL must use HTTPS protocol")
91+
verify_ssl = False
9792

9893
# Set up the OpenAPI client configuration
9994
self.config = Configuration()

cyborgdb/client/encrypted_index.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,6 @@ def train(
240240
batch_size: Size of each batch for training. Default is 2048.
241241
max_iters: Maximum iterations for training. Default is 100.
242242
tolerance: Convergence tolerance for training. Default is 1e-6.
243-
max_memory: Maximum memory (MB) usage during training. Default is 0 (no limit).
244243
245244
Note:
246245
There must be at least 2 * n_lists vector embeddings in the index prior to calling

cyborgdb/integrations/langchain.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,6 @@ def __init__(
8484
index_config_params: Optional[Dict[str, Any]] = None,
8585
dimension: Optional[int] = None,
8686
metric: str = "cosine",
87-
max_cache_size: int = 0,
8887
verify_ssl: Optional[bool] = None
8988
) -> None:
9089
"""
@@ -103,7 +102,6 @@ def __init__(
103102
index_config_params: Additional index configuration parameters
104103
dimension: Embedding dimension (auto-detected if not provided)
105104
metric: Distance metric - "cosine", "euclidean", or "squared_euclidean"
106-
max_cache_size: Maximum cache size for the index
107105
verify_ssl: SSL verification (None for auto-detect, True/False to override)
108106
109107
Raises:
@@ -113,7 +111,6 @@ def __init__(
113111

114112
self.index_name = index_name
115113
self.index_key = index_key
116-
self.max_cache_size = max_cache_size
117114

118115
# Set up embedding model
119116
self._setup_embedding_model(embedding)
@@ -171,8 +168,7 @@ def _load_existing_index(self) -> None:
171168
index_name=self.index_name,
172169
index_key=self.index_key,
173170
api=self.client.api,
174-
api_client=self.client.api_client,
175-
max_cache_size=self.max_cache_size
171+
api_client=self.client.api_client
176172
)
177173

178174
def _create_new_index(
@@ -195,8 +191,7 @@ def _create_new_index(
195191
index_name=self.index_name,
196192
index_key=self.index_key,
197193
index_config=config,
198-
embedding_model=self.embedding_model_name if self.embedding_model_name else None,
199-
max_cache_size=self.max_cache_size
194+
embedding_model=self.embedding_model_name if self.embedding_model_name else None
200195
)
201196

202197
def _detect_embedding_dimension(self) -> int:
@@ -775,7 +770,6 @@ def from_texts(
775770
index_type = kwargs.pop("index_type", "ivfflat")
776771
metric = kwargs.pop("metric", "cosine")
777772
dimension = kwargs.pop("dimension", None)
778-
max_cache_size = kwargs.pop("max_cache_size", 0)
779773
verify_ssl = kwargs.pop("verify_ssl", None)
780774

781775
# Handle index config
@@ -795,7 +789,6 @@ def from_texts(
795789
index_config_params=index_config_params,
796790
dimension=dimension,
797791
metric=metric,
798-
max_cache_size=max_cache_size,
799792
verify_ssl=verify_ssl
800793
)
801794

tests/basic_integration_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def setUpClass(cls):
137137
# CYBORDB SETUP: Create the index once (shared state).
138138
cls.index_config = cyborgdb.IndexIVFFlat(dimension=cls.dimension, n_lists=512, metric="euclidean")
139139
cls.client = cyborgdb.Client(
140-
api_url="https://localhost:8000",
140+
api_url="http://localhost:8000",
141141
api_key=os.getenv("CYBORGDB_API_KEY", "")
142142
)
143143
cls.index_name = "memory_example_index19"

tests/test_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def setUp(self):
2222
"""Set up the test environment."""
2323
# Create real client (no mocking)
2424
self.client = Client(
25-
api_url="https://localhost:8000",
25+
api_url="http://localhost:8000",
2626
api_key=os.getenv("CYBORGDB_API_KEY", "")
2727
)
2828

tests/test_langchain_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -459,7 +459,7 @@ def test_09_as_retriever(self):
459459
)
460460

461461
# Use retriever
462-
docs = retriever.get_relevant_documents("What is deep learning?")
462+
docs = retriever.invoke("What is deep learning?")
463463

464464
self.assertEqual(len(docs), 3)
465465
for doc in docs:

0 commit comments

Comments
 (0)