Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change template to private attribute and simplify imports (#101)

Browse Source 
---------

Co-authored-by: ian <ian@cinnamon.is>
This commit is contained in:
Duc Nguyen (john)
2023-12-08 18:10:34 +07:00
committed by GitHub
parent 1f927d3391
commit da0ac1d69f
13 changed files with 31 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
templates/project-default/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/pipeline.py
View File

@@ -1,10 +1,7 @@
import os
from typing import List
from theflow import Node, Param
from theflow.utils.modules import ObjectInitDeclaration as _
from kotaemon.base import BaseComponent, Document, LLMInterface
from kotaemon.base import BaseComponent, Document, LLMInterface, Node, Param, lazy
from kotaemon.contribs.promptui.logs import ResultLog
from kotaemon.embeddings import AzureOpenAIEmbeddings
from kotaemon.indices import VectorIndexing, VectorRetrieval
@@ -48,8 +45,8 @@ class QuestionAnsweringPipeline(BaseComponent):
retrieving_pipeline: VectorRetrieval = Node(
VectorRetrieval.withx(
vector_store=_(ChromaVectorStore).withx(path="./tmp"),
doc_store=_(SimpleFileDocumentStore).withx(path="docstore.json"),
vector_store=lazy(ChromaVectorStore).withx(path="./tmp"),
doc_store=lazy(SimpleFileDocumentStore).withx(path="docstore.json"),
embedding=AzureOpenAIEmbeddings.withx(
model="text-embedding-ada-002",
deployment="dummy-q2-text-embedding",
@@ -78,11 +75,11 @@ class QuestionAnsweringPipeline(BaseComponent):
class IndexingPipeline(VectorIndexing):
vector_store: ChromaVectorStore = Param(
_(ChromaVectorStore).withx(path="./tmp"),
lazy(ChromaVectorStore).withx(path="./tmp"),
ignore_ui=True,
)
doc_store: SimpleFileDocumentStore = Param(
_(SimpleFileDocumentStore).withx(path="docstore.json"),
lazy(SimpleFileDocumentStore).withx(path="docstore.json"),
ignore_ui=True,
)
embedding: AzureOpenAIEmbeddings = AzureOpenAIEmbeddings.withx(