xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-09 04:40:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: support 3072-dimensional knowledge embeddings (#2401)

Browse Source 
Co-authored-by: dadachann <185672915+dadachann@users.noreply.github.com>
This commit is contained in:
Hyu
2026-08-05 21:19:57 +08:00
committed by GitHub
parent 3b4698463c
commit cdd5c6589c
13 changed files with 95 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit_tests/cloud/test_bootstrap.py
+1 -2
View File
@@ -108,7 +108,7 @@ def _cloud_config() -> dict:
'use': 'pgvector',
'pgvector': {
'use_business_database': True,
'allowed_dimensions': [384, 768, 1536],
'allowed_dimensions': [384, 768, 1536, 3072],
},
},
'mcp': {'stdio': {'enabled': False}},
@@ -216,7 +216,6 @@ async def test_cloud_directory_capacity_contract_is_fail_closed(directory_config
[
({'use_business_database': False, 'allowed_dimensions': [1536]}, 'use_business_database=true'),
({'use_business_database': True, 'allowed_dimensions': []}, 'allowed_dimensions'),
({'use_business_database': True, 'allowed_dimensions': [3072]}, 'allowed_dimensions'),
({'use_business_database': True, 'allowed_dimensions': [True]}, 'allowed_dimensions'),
],
)
tests/unit_tests/vector/test_mgr.py
+3 -3
View File
@@ -213,7 +213,7 @@ class TestVectorDBManagerInitialization:
mock_app,
connection_string='postgresql://user:pass@host:5432/langbot',
use_business_database=False,
allowed_dimensions=[384, 512, 768, 1024, 1536],
allowed_dimensions=[384, 512, 768, 1024, 1536, 3072],
)
def test_initialize_pgvector_with_individual_params(self):
@@ -251,7 +251,7 @@ class TestVectorDBManagerInitialization:
user='admin',
password='secret',
use_business_database=False,
allowed_dimensions=[384, 512, 768, 1024, 1536],
allowed_dimensions=[384, 512, 768, 1024, 1536, 3072],
)
def test_initialize_pgvector_defaults(self):
@@ -280,7 +280,7 @@ class TestVectorDBManagerInitialization:
user='postgres',
password='postgres',
use_business_database=False,
allowed_dimensions=[384, 512, 768, 1024, 1536],
allowed_dimensions=[384, 512, 768, 1024, 1536, 3072],
)
def test_initialize_pgvector_with_shared_business_database(self):