Skip to content

D
dify
Commit 7e888698 authored by John Wang's avatar John Wang
Browse files
Options

Merge branch 'feat/claude-api-support' into deploy/dev

parents d6361366 f840cea9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
api/core/embedding/cached_embedding.py
View file @ 7e888698
...@@ -4,6 +4,7 @@ from typing import List ...@@ -4,6 +4,7 @@ from typing import List
from langchain.embeddings.base import Embeddings from langchain.embeddings.base import Embeddings
from sqlalchemy.exc import IntegrityError from sqlalchemy.exc import IntegrityError
from core.llm.wrappers.openai_wrapper import handle_openai_exceptions
from extensions.ext_database import db from extensions.ext_database import db
from libs import helper from libs import helper
from models.dataset import Embedding from models.dataset import Embedding
...@@ -49,6 +50,7 @@ class CacheEmbedding(Embeddings): ...@@ -49,6 +50,7 @@ class CacheEmbedding(Embeddings):
text_embeddings.extend(embedding_results) text_embeddings.extend(embedding_results)
return text_embeddings return text_embeddings
@handle_openai_exceptions
def embed_query(self, text: str) -> List[float]: def embed_query(self, text: str) -> List[float]:
"""Embed query text.""" """Embed query text."""
# use doc embedding cache or store if not exists # use doc embedding cache or store if not exists
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment