Update docs (#106)
This commit is contained in:
@@ -4,10 +4,11 @@ from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
from index import ReaderIndexingPipeline
|
||||
from kotaemon.llms import AzureChatOpenAI
|
||||
from openai.resources.embeddings import Embeddings
|
||||
from openai.types.chat.chat_completion import ChatCompletion
|
||||
|
||||
from kotaemon.llms import AzureChatOpenAI
|
||||
|
||||
with open(Path(__file__).parent / "resources" / "embedding_openai.json") as f:
|
||||
openai_embedding = json.load(f)
|
||||
|
||||
|
@@ -3,11 +3,12 @@ import logging
|
||||
from functools import cache
|
||||
from pathlib import Path
|
||||
|
||||
from kotaemon.base import BaseComponent
|
||||
from kotaemon.storages import BaseDocumentStore, BaseVectorStore
|
||||
from theflow.settings import settings
|
||||
from theflow.utils.modules import deserialize
|
||||
|
||||
from kotaemon.base import BaseComponent
|
||||
from kotaemon.storages import BaseDocumentStore, BaseVectorStore
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
|
@@ -17,10 +17,6 @@ from ktem.components import (
|
||||
from ktem.db.models import Index, Source, SourceTargetRelation, engine
|
||||
from ktem.indexing.base import BaseIndexing, BaseRetriever
|
||||
from ktem.indexing.exceptions import FileExistsError
|
||||
from kotaemon.base import RetrievedDocument
|
||||
from kotaemon.indices import VectorIndexing, VectorRetrieval
|
||||
from kotaemon.indices.ingests import DocumentIngestor
|
||||
from kotaemon.indices.rankings import BaseReranking, CohereReranking, LLMReranking
|
||||
from llama_index.vector_stores import (
|
||||
FilterCondition,
|
||||
FilterOperator,
|
||||
@@ -31,6 +27,11 @@ from llama_index.vector_stores.types import VectorStoreQueryMode
|
||||
from sqlmodel import Session, select
|
||||
from theflow.settings import settings
|
||||
|
||||
from kotaemon.base import RetrievedDocument
|
||||
from kotaemon.indices import VectorIndexing, VectorRetrieval
|
||||
from kotaemon.indices.ingests import DocumentIngestor
|
||||
from kotaemon.indices.rankings import BaseReranking, CohereReranking, LLMReranking
|
||||
|
||||
USER_SETTINGS = {
|
||||
"index_parser": {
|
||||
"name": "Index parser",
|
||||
|
@@ -6,6 +6,7 @@ from functools import partial
|
||||
import tiktoken
|
||||
from ktem.components import llms
|
||||
from ktem.indexing.base import BaseRetriever
|
||||
|
||||
from kotaemon.base import (
|
||||
BaseComponent,
|
||||
Document,
|
||||
|
Reference in New Issue
Block a user