Skip to content

Commit 18b5dad

Browse files
authored
fix: avoid duplicate dynamic model creation (#684)
1 parent aa715ba commit 18b5dad

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

backend/app/models/chunk.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import enum
22
from functools import lru_cache
33

4+
import sys
45
from typing import Optional, Type
56
from sqlmodel import (
67
Field,
@@ -34,7 +35,7 @@ def get_kb_chunk_model(kb: KnowledgeBase) -> Type[SQLModel]:
3435
return get_dynamic_chunk_model(vector_dimension, str(kb.id))
3536

3637

37-
@lru_cache(maxsize=None)
38+
@lru_cache(maxsize=sys.maxsize)
3839
def get_dynamic_chunk_model(
3940
vector_dimension: int,
4041
namespace: Optional[str] = None,

backend/app/models/entity.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import enum
22
from functools import lru_cache
3+
import sys
34
from typing import Optional, List, Dict, Type
45

56
from sqlmodel import (
@@ -40,7 +41,7 @@ def get_kb_entity_model(kb: KnowledgeBase) -> Type[SQLModel]:
4041
return get_dynamic_entity_model(vector_dimension, str(kb.id))
4142

4243

43-
@lru_cache(maxsize=None)
44+
@lru_cache(maxsize=sys.maxsize)
4445
def get_dynamic_entity_model(
4546
vector_dimension: int,
4647
namespace: Optional[str] = None,

backend/app/models/relationship.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from datetime import datetime
22
from functools import lru_cache
3+
import sys
34
from typing import Optional, Type
45
from uuid import UUID
56

@@ -36,7 +37,7 @@ def get_kb_relationship_model(kb: KnowledgeBase) -> Type[SQLModel]:
3637
return get_dynamic_relationship_model(vector_dimension, str(kb.id), entity_model)
3738

3839

39-
@lru_cache(maxsize=None)
40+
@lru_cache(maxsize=sys.maxsize)
4041
def get_dynamic_relationship_model(
4142
vector_dimension: int,
4243
namespace: Optional[str] = None,

0 commit comments

Comments
 (0)