Skip to content

Commit d040732

Browse files
committed
[fix] don't use tiktoken by default with langchain
1 parent 1e62fcc commit d040732

File tree

5 files changed

+92
-2
lines changed

5 files changed

+92
-2
lines changed

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ dependencies = [
1515
"fastapi>=0.115.12",
1616
"streamlit>=1.44.0",
1717
"langchain-openai>=0.3.11",
18+
"transformers>=4.51.3",
1819
]
1920
authors = [
2021
{name="Thomas Faria", email="[email protected]"}

src/constants/graph_db.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@
44
NEO4J_USERNAME = "neo4j"
55
NEO4J_PWD = os.environ["NEO4J_API_KEY"]
66
EMBEDDING_MODEL = "ordalieTech/Solon-embeddings-large-0.1"
7-
URL_EMBEDDING_API = "https://projet-ape-vllm-embed.user.lab.sspcloud.fr/v1"
7+
URL_EMBEDDING_API = "https://vllm-embed.user.lab.sspcloud.fr/v1"

src/constants/llm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
GENERATION_MODEL = "mistralai/Mistral-Small-24B-Instruct-2501"
2-
URL_LLM_API = "https://projet-ape-vllm.user.lab.sspcloud.fr/v1"
2+
URL_LLM_API = "https://vllm-generation.user.lab.sspcloud.fr/v1"

src/vector_db/loaders.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ def get_embedding_model(model_name: str) -> OpenAIEmbeddings:
3232
model=model_name,
3333
openai_api_base=URL_EMBEDDING_API,
3434
openai_api_key="EMPTY",
35+
tiktoken_enabled=False,
3536
)
3637

3738

uv.lock

Lines changed: 88 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)