Hugging Face's logo

aiflows
/
VectorStoreFlowModule

Model card Files
xet
Community
VectorStoreFlowModule
File size: 2,792 Bytes 
c415e05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
chroma_demo_flow:
  input_interface:
    - "operation"
    - "content"
  output_interface:
    - "retrieved"
  _target_: flows.base_flows.SequentialFlow.instantiate_from_default_config
  name: "demoChromaDBFlow"
  description: "An example flow of how to read and writed in a ChromaDBFlowModule."
  subflows_config:
    chroma_db:
      input_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_select: ["operation","content"]
      _target_: aiflows.VectorStoreFlowModule.ChromaDBFlow.instantiate_from_default_config
      
      backend:
        _target_: flows.backends.llm_lite.LiteLLMBackend
        api_infos: ???
        model_name: "" #Not used in current implementation
      n_results: 1 # number of results to retrieve when query
  topology:
    - goal: Write content to the ChromaDB
      input_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_select: ["operation","content"]
      flow: chroma_db
      output_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_set:
            operation: "read"
        keys_to_rename:
            retrieved: content
        keys_to_select: ["operation","content"]

    - goal: Read content from the ChromaDB
      input_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_select: ["operation","content"]
      flow: chroma_db
      output_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_select: ["retrieved"]

vector_store_demo_flow:
  input_interface:
      - "operation"
      - "content"
  output_interface:
    - "retrieved"
  name: "demoVectorStoreFlow"
  description: "An example flow of how to read and write in a VectorStoreFlowModule."
  _target_: flows.base_flows.SequentialFlow.instantiate_from_default_config
  subflows_config:
    
    vs_db:
      _target_: aiflows.VectorStoreFlowModule.VectorStoreFlow.instantiate_from_default_config
      backend:
        _target_: flows.backends.llm_lite.LiteLLMBackend
        api_infos: ???
        model_name: "" #Not used in current implementation

  topology:
    - goal: Write content to the VectorStore
      input_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_select: ["operation","content"]
      flow: vs_db
      output_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_set:
            operation: "read"
        keys_to_rename:
            retrieved: content
        keys_to_select: ["operation","content"]

    - goal: Read content from the VectorStore
      input_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_select: ["operation","content"]
      flow: vs_db
      output_interface:
        _target_: flows.interfaces.KeyInterface
        keys_to_select: ["retrieved"]