lfx-nightly 0.1.13.dev2__py3-none-any.whl → 0.1.13.dev4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of lfx-nightly might be problematic. Click here for more details.

Files changed (67) hide show
  1. lfx/_assets/component_index.json +1 -1
  2. lfx/base/agents/agent.py +17 -1
  3. lfx/base/agents/utils.py +15 -2
  4. lfx/base/composio/composio_base.py +24 -9
  5. lfx/base/datastax/__init__.py +5 -0
  6. lfx/{components/vectorstores/astradb.py → base/datastax/astradb_base.py} +84 -473
  7. lfx/base/io/chat.py +5 -4
  8. lfx/base/mcp/util.py +101 -15
  9. lfx/cli/commands.py +1 -1
  10. lfx/components/agents/agent.py +1 -1
  11. lfx/components/agents/cuga_agent.py +1 -1
  12. lfx/components/agents/mcp_component.py +16 -0
  13. lfx/components/amazon/amazon_bedrock_converse.py +1 -1
  14. lfx/components/apify/apify_actor.py +3 -3
  15. lfx/components/datastax/__init__.py +12 -6
  16. lfx/components/datastax/{astra_assistant_manager.py → astradb_assistant_manager.py} +1 -0
  17. lfx/components/datastax/astradb_chatmemory.py +40 -0
  18. lfx/components/datastax/astradb_cql.py +5 -31
  19. lfx/components/datastax/astradb_graph.py +9 -123
  20. lfx/components/datastax/astradb_tool.py +12 -52
  21. lfx/components/datastax/astradb_vectorstore.py +133 -976
  22. lfx/components/datastax/create_assistant.py +1 -0
  23. lfx/components/datastax/create_thread.py +1 -0
  24. lfx/components/datastax/dotenv.py +1 -0
  25. lfx/components/datastax/get_assistant.py +1 -0
  26. lfx/components/datastax/getenvvar.py +1 -0
  27. lfx/components/datastax/graph_rag.py +1 -1
  28. lfx/components/datastax/list_assistants.py +1 -0
  29. lfx/components/datastax/run.py +1 -0
  30. lfx/components/knowledge_bases/ingestion.py +17 -9
  31. lfx/components/knowledge_bases/retrieval.py +16 -8
  32. lfx/components/mistral/mistral_embeddings.py +1 -1
  33. lfx/components/openrouter/openrouter.py +49 -147
  34. lfx/components/vectorstores/__init__.py +0 -6
  35. lfx/custom/custom_component/component.py +3 -2
  36. lfx/graph/edge/base.py +2 -2
  37. lfx/graph/graph/base.py +1 -1
  38. lfx/graph/graph/schema.py +3 -2
  39. lfx/graph/vertex/vertex_types.py +1 -1
  40. {lfx_nightly-0.1.13.dev2.dist-info → lfx_nightly-0.1.13.dev4.dist-info}/METADATA +1 -1
  41. {lfx_nightly-0.1.13.dev2.dist-info → lfx_nightly-0.1.13.dev4.dist-info}/RECORD +44 -65
  42. lfx/components/datastax/astra_db.py +0 -77
  43. lfx/components/datastax/cassandra.py +0 -92
  44. lfx/components/vectorstores/astradb_graph.py +0 -326
  45. lfx/components/vectorstores/cassandra.py +0 -264
  46. lfx/components/vectorstores/cassandra_graph.py +0 -238
  47. lfx/components/vectorstores/chroma.py +0 -167
  48. lfx/components/vectorstores/clickhouse.py +0 -135
  49. lfx/components/vectorstores/couchbase.py +0 -102
  50. lfx/components/vectorstores/elasticsearch.py +0 -267
  51. lfx/components/vectorstores/faiss.py +0 -111
  52. lfx/components/vectorstores/graph_rag.py +0 -141
  53. lfx/components/vectorstores/hcd.py +0 -314
  54. lfx/components/vectorstores/milvus.py +0 -115
  55. lfx/components/vectorstores/mongodb_atlas.py +0 -213
  56. lfx/components/vectorstores/opensearch.py +0 -243
  57. lfx/components/vectorstores/pgvector.py +0 -72
  58. lfx/components/vectorstores/pinecone.py +0 -134
  59. lfx/components/vectorstores/qdrant.py +0 -109
  60. lfx/components/vectorstores/supabase.py +0 -76
  61. lfx/components/vectorstores/upstash.py +0 -124
  62. lfx/components/vectorstores/vectara.py +0 -97
  63. lfx/components/vectorstores/vectara_rag.py +0 -164
  64. lfx/components/vectorstores/weaviate.py +0 -89
  65. /lfx/components/datastax/{astra_vectorize.py → astradb_vectorize.py} +0 -0
  66. {lfx_nightly-0.1.13.dev2.dist-info → lfx_nightly-0.1.13.dev4.dist-info}/WHEEL +0 -0
  67. {lfx_nightly-0.1.13.dev2.dist-info → lfx_nightly-0.1.13.dev4.dist-info}/entry_points.txt +0 -0
@@ -4,17 +4,17 @@ lfx/constants.py,sha256=Ert_SpwXhutgcTKEvtDArtkONXgyE5x68opMoQfukMA,203
4
4
  lfx/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
5
  lfx/settings.py,sha256=wnx4zkOLQ8mvampYsnnvVV9GvEnRUuWQpKFSbFTCIp4,181
6
6
  lfx/type_extraction.py,sha256=eCZNl9nAQivKdaPv_9BK71N0JV9Rtr--veAht0dnQ4A,2921
7
- lfx/_assets/component_index.json,sha256=zmPfN3r3mfThSSgHe0kk_cH59hjk2-GReaVZjP6cfcw,3913662
7
+ lfx/_assets/component_index.json,sha256=9evw0dFyNLrnmQYj_AJLTTFf-MJajP1iU7k-Qwqtf2Y,3525471
8
8
  lfx/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
9
9
  lfx/base/constants.py,sha256=v9vo0Ifg8RxDu__XqgGzIXHlsnUFyWM-SSux0uHHoz8,1187
10
10
  lfx/base/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
- lfx/base/agents/agent.py,sha256=-PZ0gzBCXB4EXDHoXhYNY-tPvcOogfSNfV_eb1B1u-c,11784
11
+ lfx/base/agents/agent.py,sha256=daiAVaVCSsg3VC3HMVUUrOaLTi6BkZNv5xRB6OIG_4k,12695
12
12
  lfx/base/agents/callback.py,sha256=mjlT9ukBMVrfjYrHsJowqpY4g9hVGBVBIYhncLWr3tQ,3692
13
13
  lfx/base/agents/context.py,sha256=u0wboX1aRR22Ia8gY14WF12RjhE0Rxv9hPBiixT9DtQ,3916
14
14
  lfx/base/agents/default_prompts.py,sha256=tUjfczwt4D5R1KozNOl1uSL2V2rSCZeUMS-cfV4Gwn0,955
15
15
  lfx/base/agents/errors.py,sha256=4QY1AqSWZaOjq-iQRYH_aeCfH_hWECLQkiwybNXz66U,531
16
16
  lfx/base/agents/events.py,sha256=jOhfGf_za5IYMvxRMvi1qqU0gw9b1imvDCOQP3r3wlo,14598
17
- lfx/base/agents/utils.py,sha256=OcmtZx4BTFTyq2A3rta3WoJn98UzEYdEXoRLs8-mTVo,6511
17
+ lfx/base/agents/utils.py,sha256=VEAVYC6oOadjKeZ-cUS-1OOCnWW69FhpcGWzjvR4uZ8,7161
18
18
  lfx/base/agents/crewai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
19
19
  lfx/base/agents/crewai/crew.py,sha256=TN1JyLXMpJc2yPH3tokhFmxKKYoJ4lMvmG19DmpKfeY,7953
20
20
  lfx/base/agents/crewai/tasks.py,sha256=1pBok1UDdAjLtOf2Y-rDrjRaM93no-XLy5Bf_zvWsRM,151
@@ -23,7 +23,7 @@ lfx/base/astra_assistants/util.py,sha256=T_W44VFoOXBF3m-0eCSrHvzbKx1gdyBF9IAWKMX
23
23
  lfx/base/chains/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
24
24
  lfx/base/chains/model.py,sha256=QSYJBc0Ygpx2Ko273u1idL_gPK2xpvRQgJb4oTx8x8s,766
25
25
  lfx/base/composio/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
26
- lfx/base/composio/composio_base.py,sha256=AJBitxwJAirGm000njuLBdD_uBHKbcPvfHDyXuJdViY,131914
26
+ lfx/base/composio/composio_base.py,sha256=3mcseDIDIj25oKDsTnnelvSq8dIImzrFJDmkcNSWOy0,132478
27
27
  lfx/base/compressors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
28
  lfx/base/compressors/model.py,sha256=-FFBAPAy9bAgvklIo7x_uwShZR5NoMHakF6f_hNnLHg,2098
29
29
  lfx/base/curl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -32,6 +32,8 @@ lfx/base/data/__init__.py,sha256=lQsYYMyAg_jA9ZF7oc-LNZsRE2uMGT6g16WzsUByHqs,81
32
32
  lfx/base/data/base_file.py,sha256=UKpF9BsNHgg-cdB1uVB8F00czvCTLfks320aLg3F_kM,28049
33
33
  lfx/base/data/docling_utils.py,sha256=gVDxOZghSJEo5n-UNkVGBQYqkvfNqkNkltBhAnoaJd4,13048
34
34
  lfx/base/data/utils.py,sha256=dGqEO4zE5s_V2Cs4j0EEeyLjYLX6Zex-EGzIOznK76o,5960
35
+ lfx/base/datastax/__init__.py,sha256=s72q8NeqlMuNXb5WcYR2DTIuWOiBtfX23Z_Zte2PHGo,90
36
+ lfx/base/datastax/astradb_base.py,sha256=-6ROQSZtJ1tBXH6984bau-O4nl7NzB8C7Sz9T81YCdI,36776
35
37
  lfx/base/document_transformers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
36
38
  lfx/base/document_transformers/model.py,sha256=etVEmyakiEgflB-fayClPnFRhaEdXfdUu4cqpgtk8ek,1317
37
39
  lfx/base/embeddings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,7 +44,7 @@ lfx/base/flow_processing/utils.py,sha256=G-MhVp_W9xdNVYrajovky31bNWgUqq5H4GDlEDW
42
44
  lfx/base/huggingface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
43
45
  lfx/base/huggingface/model_bridge.py,sha256=pObEcu70zRdSZItl9P5SfxdpxTQJeE9Sp79ISbGcoy0,4800
44
46
  lfx/base/io/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
45
- lfx/base/io/chat.py,sha256=SUIp_I_Xa2BhdfoFMYppob0_4KjRAjebtN01Vm0cXm4,1007
47
+ lfx/base/io/chat.py,sha256=9FjHtd8Qjl8TqNSCuij5LSO1HGs0FNreGFBZa8F-Edw,1016
46
48
  lfx/base/io/text.py,sha256=uW2vdNIccZUzv0FUVafo4W-lOhm8YFJOMV2qytGYHPk,748
47
49
  lfx/base/knowledge_bases/__init__.py,sha256=qn65fdRRbwoD1qxHd45sze1ylESqPJaq76TvBysLXbg,151
48
50
  lfx/base/knowledge_bases/knowledge_base_utils.py,sha256=yxYm5DTk-TPkibFH_z44jql_MA21y_HaNbk7WZ3eW1Q,4399
@@ -53,7 +55,7 @@ lfx/base/langwatch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
53
55
  lfx/base/langwatch/utils.py,sha256=N7rH3sRwgmNQzG0pKjj4wr_ans_drwtvkx4BQt-B0WA,457
54
56
  lfx/base/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
55
57
  lfx/base/mcp/constants.py,sha256=-1XnJxejlqm9zs1R91qGtOeX-_F1ZpdHVzCIqUCvhgE,62
56
- lfx/base/mcp/util.py,sha256=rHi-wC8tuUWCiI65AylJNE7dmO59K71R_W-ruN3MO50,69164
58
+ lfx/base/mcp/util.py,sha256=PpXfEpZTzPcwLXqKimBNh37k3s21wRErxKnNXIUd980,71844
57
59
  lfx/base/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
58
60
  lfx/base/memory/memory.py,sha256=kZ-aZoHvRW4PJAgY1LUt5UBj7YXbos_aVBPGjC1EFCY,1835
59
61
  lfx/base/memory/model.py,sha256=2oDORZV_l-DHLx9j9--wYprQUIYKOb8aTJpXmR1qOLw,1330
@@ -91,7 +93,7 @@ lfx/base/vectorstores/model.py,sha256=pDAZ6D6XnMxGAV9hJjc3DYhjI9n77sc_FIs5lnpsDb
91
93
  lfx/base/vectorstores/utils.py,sha256=OhBNYs9Z9poe82rTNFPdrESNRGuP6RO6-eOpwqJLBG0,750
92
94
  lfx/base/vectorstores/vector_store_connection_decorator.py,sha256=2gh3DMhcMsCgVYFEFaVNMT3zsbd-fkFy5Bl_-jXDu8c,1998
93
95
  lfx/cli/__init__.py,sha256=Oy17zrnwBlwJn80sMGyRJXos2n2eQGvSsh9CS_-v2R4,113
94
- lfx/cli/commands.py,sha256=IR4leAtfQUjsc9wI6q1LRx--u4sW-i5ixtKdMYju-hI,11912
96
+ lfx/cli/commands.py,sha256=kDqZj78QE0tcPCB-k8pTy6D_nRUFkUW2lsFGn7GQlU0,11898
95
97
  lfx/cli/common.py,sha256=SvszBhWoOttM27rButhBOpvE8lO5UGMYL0NaG_OW8bc,22060
96
98
  lfx/cli/run.py,sha256=U4oKC21-jOu1mBTkWI4aASnvKuzi_ak3egZEci51ofU,21593
97
99
  lfx/cli/script_loader.py,sha256=xWSpx57cBeX0UHmUgAk97aye9-hhD2Y6nKh68A-xaTA,8997
@@ -113,21 +115,21 @@ lfx/components/Notion/update_page_property.py,sha256=tgmPMbD1eX58dQQNXv1w5FzDec7
113
115
  lfx/components/agentql/__init__.py,sha256=Erl669Dzsk-SegsDPWTtkKbprMXVuv8UTCo5REzZGTc,56
114
116
  lfx/components/agentql/agentql_api.py,sha256=N94yEK7ZuQCIsFBlr_8dqrJY-K1-KNb6QEEYfDIsDME,5569
115
117
  lfx/components/agents/__init__.py,sha256=UBu5kO9hp8yFyxTU-u9KHN9zTSoHhJSYdKtRuT5ig9c,1164
116
- lfx/components/agents/agent.py,sha256=vbi422N1dpML7fpYn3FqcwQ5nYS7BQBnmSN--WnheOg,26988
117
- lfx/components/agents/cuga_agent.py,sha256=r_342LveLdERtvE84UJJOdKTEwWSIjvASRYke9JV2Ik,44494
118
- lfx/components/agents/mcp_component.py,sha256=mE2HvbHcdkuWWylxmaNNZobbtgBRktOOakeGwUYs7Qs,25586
118
+ lfx/components/agents/agent.py,sha256=-pGSOYS-2pHoYq5VyNdDCnkQVzVz52as4hdF_BB1hVI,26993
119
+ lfx/components/agents/cuga_agent.py,sha256=JNi4MsSziTJQI3z_0KGzNWxm5RDaMk_W9zcTW2KcTtI,44499
120
+ lfx/components/agents/mcp_component.py,sha256=I8xAV7qGy67uV_4R6lrM4f6EOa9cFPQZDIcP3tiMuYk,26208
119
121
  lfx/components/aiml/__init__.py,sha256=DNKB-HMFGFYmsdkON-s8557ttgBXVXADmS-BcuSQiIQ,1087
120
122
  lfx/components/aiml/aiml.py,sha256=23Ineg1ajlCoqXgWgp50I20OnQbaleRNsw1c6IzPu3A,3877
121
123
  lfx/components/aiml/aiml_embeddings.py,sha256=2uNwORuj55mxn2SfLbh7oAIfjuXwHbsnOqRjfMtQRqc,1095
122
124
  lfx/components/amazon/__init__.py,sha256=QRsw-7ytuqZWhNPmlslCQEcpNsAw6dX5FV_8ihUdlPQ,1338
123
- lfx/components/amazon/amazon_bedrock_converse.py,sha256=WPo_5Wsj-pyltXcBOp6xPmtwVzRcJ5JOz5W013Ev9EE,8190
125
+ lfx/components/amazon/amazon_bedrock_converse.py,sha256=VMM1-Gmf-XUfaW-LChb4DnFUF_5LfA1LIo84kmjvUGE,8195
124
126
  lfx/components/amazon/amazon_bedrock_embedding.py,sha256=cNA5_3nwkegWS3BY75IfjOwpo-g8dpg-EgtQJSgVXfg,4261
125
127
  lfx/components/amazon/amazon_bedrock_model.py,sha256=kiCTqDG2Krbe0i5YBG9UODxwNTB9ePZwLfgNMR91UwQ,5113
126
128
  lfx/components/amazon/s3_bucket_uploader.py,sha256=bkui2vw8ibgg4dSUYZ7yAEbVOjgmB8oOYqH-3ZvTlBQ,7688
127
129
  lfx/components/anthropic/__init__.py,sha256=SGllKMRoX3KQkDpdzsLUYqbfTbhFu9qjpM6bQHL2syQ,965
128
130
  lfx/components/anthropic/anthropic.py,sha256=fIlMWma6rsfQ499vkkLSWFZqwKn5ifvhFQnC_quHkEw,7289
129
131
  lfx/components/apify/__init__.py,sha256=ADNTjMjTglvdC34YEgMYWippzrXWiH3ku_3RvRQcOlw,89
130
- lfx/components/apify/apify_actor.py,sha256=O8au5opTgLuv5QY-OUzV4-Pst1clxGKWgHPm08i5PFQ,12853
132
+ lfx/components/apify/apify_actor.py,sha256=n8_bD7CAbGG-6FQrUYFlZGlQzl71wGXuOqHp-bHEoYc,12850
131
133
  lfx/components/arxiv/__init__.py,sha256=g3uhahiWz1ZALR65oBFfO-hmrHcJLb5HuUqbeTFZ2Ss,64
132
134
  lfx/components/arxiv/arxiv.py,sha256=IZI57itI_oiu_BA3q9QQRP8bh2TqIjrcOXCR7QT_DSM,6627
133
135
  lfx/components/assemblyai/__init__.py,sha256=H0Rt2gzDDgj6IL8laOpolIFzhueT08VaMiVetGvlt0c,1632
@@ -239,24 +241,23 @@ lfx/components/data/sql_executor.py,sha256=sN1lWM65O_pCfZxNAzgjtZmcTPGBLqMud2_7n
239
241
  lfx/components/data/url.py,sha256=zbfTeTBukw3F_mRBMIJrQYF94psEIBuS8dFjcQku5SE,11001
240
242
  lfx/components/data/web_search.py,sha256=48SCp-2I_Qckp5tmTVC9JBw2C-MhBDF14MJLaGjLpyQ,12758
241
243
  lfx/components/data/webhook.py,sha256=i2jdXSLUVA0UpnYBZzdPo035MeiUcFKVJy37EhLKq6o,1643
242
- lfx/components/datastax/__init__.py,sha256=VEh_Qu8dYPOVB9dISRaxheFPKxzQoNNe1DCwTWTGNIU,2415
243
- lfx/components/datastax/astra_assistant_manager.py,sha256=5vLbuCxSz04GKxEpP0TNm9K_RAxxoMTz-Mt_YhkyfH0,11557
244
- lfx/components/datastax/astra_db.py,sha256=i5LL87sirODcFEugJVjs0PPgk-9yCyz8lwMQFynrqvM,2788
245
- lfx/components/datastax/astra_vectorize.py,sha256=PZdmkMJiPAnTankXjFjdG192OeHVetxBfGzbwxgPQ54,4912
246
- lfx/components/datastax/astradb_cql.py,sha256=pHms51vCfx-h0Vyjhoi0Sr3WLVWww9XEyeuQVZTxy84,11774
247
- lfx/components/datastax/astradb_graph.py,sha256=Og-9UMxWk3CzEJ2-fGPnfLzik3dzlbIDEKMYvd3ZMM0,12843
248
- lfx/components/datastax/astradb_tool.py,sha256=5tIrdsxbsZ8NpmxBHww_5_tDnCr-Me5kUtJRxxSQT4Y,16744
249
- lfx/components/datastax/astradb_vectorstore.py,sha256=K2OnNfuJfEGMwnkLgIzh8gKWMFbDpOVBUntSt9tqPO0,53710
250
- lfx/components/datastax/cassandra.py,sha256=9klxgphOLxOPu32cweSt6UJXzFctJxf38v1nGhumpNo,3197
251
- lfx/components/datastax/create_assistant.py,sha256=VJ0-XMLq5otLPZNS69X5PhVfUHfEVNnkwYWx2XCLhB4,2106
252
- lfx/components/datastax/create_thread.py,sha256=zpFOpUfz_lXhtA4n6B1T0xzIYr0bhpn9SiZDiaaa3gU,1085
253
- lfx/components/datastax/dotenv.py,sha256=S7m1a5UcJ-hQvOf7-ikeI-V430fSyWxLkN_NOvs1y10,1098
254
- lfx/components/datastax/get_assistant.py,sha256=oA5HcSr3riI6xok9c_09IG-FFLUNt3DTjc0PbLUKhxY,1255
255
- lfx/components/datastax/getenvvar.py,sha256=rbg3ly45OmQfPnF2SIWp_JduAzH5U1MAXMZjkfeDJpI,947
256
- lfx/components/datastax/graph_rag.py,sha256=4NmYQkjSru_zaDhJfxdaYtap-RMGJfv2AYN2NEYSdds,5163
244
+ lfx/components/datastax/__init__.py,sha256=4zuJU2d0ucI2sCrfXQ06lpURgli_8DkIyd1szjkx2Ts,2712
245
+ lfx/components/datastax/astradb_assistant_manager.py,sha256=jUrBzuN6ws_KuWG4NplR0Mbcw-7FlDCnXFeWpckvNdc,11575
246
+ lfx/components/datastax/astradb_chatmemory.py,sha256=uvyB94x2O_6LH7CBeNp6watpYU31_wTPPPcSeXyF3t8,1475
247
+ lfx/components/datastax/astradb_cql.py,sha256=MQthGbCvoNl2ccqtiXN97T9D9yVlsqteGKNkc7LpFA4,10930
248
+ lfx/components/datastax/astradb_graph.py,sha256=-LQlwfU1NI2Pnzm4HWNmf0UonuCogzwpI3qNX32axvc,8415
249
+ lfx/components/datastax/astradb_tool.py,sha256=wMbRqH2Ovj8qTr4py2Vj4i1nmehX4FWLjkynYYoiF8g,15048
250
+ lfx/components/datastax/astradb_vectorize.py,sha256=PZdmkMJiPAnTankXjFjdG192OeHVetxBfGzbwxgPQ54,4912
251
+ lfx/components/datastax/astradb_vectorstore.py,sha256=KxLNuWtnar7NNh3DFIdFmcTYbKcj7ZfLSemaTWUnUJc,18869
252
+ lfx/components/datastax/create_assistant.py,sha256=jZhp2aid7fxH7yrDFOECmqVZ-56g4KAY_NUTIJvNN1o,2124
253
+ lfx/components/datastax/create_thread.py,sha256=XUCnOsz98fVRuTgoGgD5aw6KChIeomM1ikeXfc-zGaM,1103
254
+ lfx/components/datastax/dotenv.py,sha256=ND6pqsobuZXUg6dYJiHkrZmwWEl5bJl0gApsTVWg5J8,1116
255
+ lfx/components/datastax/get_assistant.py,sha256=H2DaFh_T3s-3P96Qg0-anBXZcpz26tRy-KO8fE6CC5A,1273
256
+ lfx/components/datastax/getenvvar.py,sha256=CD8KlPOAWZ2wqLY6M33mmlI3THvl3Q2PIvoq_gkhtTk,965
257
+ lfx/components/datastax/graph_rag.py,sha256=TYNwml9ftIGiybIemxVRPiuKBAJussW6Bi7bCUgvVlo,5162
257
258
  lfx/components/datastax/hcd.py,sha256=fnRp4DvFTHqDlKZD-a0xZBGoPrVvxaZ_vXJXzHzt-ts,12478
258
- lfx/components/datastax/list_assistants.py,sha256=W5ROMSBoZdGo4ShB9i6Uzb3lXz-hy9gr0t0N2cgUTPM,928
259
- lfx/components/datastax/run.py,sha256=yPP8_5wBsHjo0OyPYGWJ2uvgQGT446ZsoN-DF6DaaU4,3063
259
+ lfx/components/datastax/list_assistants.py,sha256=F-nFx4pu6hPK3Ka5H3Lff3ZA_9jkVWl94v_bFZhD6QA,946
260
+ lfx/components/datastax/run.py,sha256=XiGc0pDTaHPMOf8cioYfSPDVRh0d3ee3qV5iogjpiCI,3081
260
261
  lfx/components/deactivated/__init__.py,sha256=A94MZ_EW-Ckp2sgDMiXNrptAaUzMbgbkl6yVpBjJXm8,485
261
262
  lfx/components/deactivated/amazon_kendra.py,sha256=y2klL_LZCKphRQoZbimTB3P1ohUWtBuFZaTCMBnI6tg,2152
262
263
  lfx/components/deactivated/chat_litellm_model.py,sha256=8YGuEPxhRJKQ8py0y-dBSZaU6_ad5DAMhmuM5xU8p9Y,4982
@@ -357,8 +358,8 @@ lfx/components/jigsawstack/text_to_sql.py,sha256=MKoMwV-4QjDLfEs0o_HwZDJYbkhGY_1
357
358
  lfx/components/jigsawstack/text_translate.py,sha256=RiM-GZYf3oRBKaq7FkzssL2sDU9y7jf3QkMDkER-h8Y,2826
358
359
  lfx/components/jigsawstack/vocr.py,sha256=pESRlHunJ8u1QlmG3672Zomf6AeyeF1NZ5vtmDx06AQ,4158
359
360
  lfx/components/knowledge_bases/__init__.py,sha256=lcxAelAmtjSo1zEKjtCayPFasuOAIzpKjX_wgW-fTJg,1143
360
- lfx/components/knowledge_bases/ingestion.py,sha256=hJCUZn9vItzPZ0IQqaoUBOaU3E7tstt3SYQAIPlq27w,28019
361
- lfx/components/knowledge_bases/retrieval.py,sha256=v5aQLdzH9R3W_grT3bAXwCPxTP523XYq3TKAAoyzNbM,10306
361
+ lfx/components/knowledge_bases/ingestion.py,sha256=x1PpImGucmAPUDnusWCJW1uPuhvdfQ0NWq06uyO5tSQ,28385
362
+ lfx/components/knowledge_bases/retrieval.py,sha256=GJis1t9LZ7tw3zsSebV3RW2GA2UCUE6Bs9AQx9UZRlg,10665
362
363
  lfx/components/langchain_utilities/__init__.py,sha256=f39oVjaQixPLfshFl6EeI6rDZgq6U3cVQEi9C-u6Ybw,4386
363
364
  lfx/components/langchain_utilities/character.py,sha256=2jCYk9ZOiHeOLCST4GW4ojTgDa-uAFIjSJ3hdx2sTjA,1747
364
365
  lfx/components/langchain_utilities/conversation.py,sha256=sOQETT8IHogc_xZu4PIEKCuCknV0FtAxOfMmc5g2XK4,1864
@@ -411,7 +412,7 @@ lfx/components/milvus/__init__.py,sha256=ZNV3umCFDejy7MhaKOfp2M-LtJnQBY1rXbRDRDC
411
412
  lfx/components/milvus/milvus.py,sha256=K17-MTsx3cAVLj_5v4e_YJ3UTnS4SdJlQirKp22X1hw,4402
412
413
  lfx/components/mistral/__init__.py,sha256=EABXqA45Tz50vZRmhEisbIIPEcRCvV9j-Y9Hf2XevHs,1094
413
414
  lfx/components/mistral/mistral.py,sha256=4heAlIFEeq_ljUZDPpNGyK_VRkWjwCfPbBaQK1mV4NY,3718
414
- lfx/components/mistral/mistral_embeddings.py,sha256=NNBGFIocnWpYehCalxh8Csun-qdHL5J-IzPsFe5Mlv0,1974
415
+ lfx/components/mistral/mistral_embeddings.py,sha256=QuqS_S3yHWCacs-Nc3qalpUsb-OACRWFZenUtCD_rLQ,1963
415
416
  lfx/components/models/__init__.py,sha256=hhfj70MkcRATzAjJnntAg1A4E7kHlQn8GT0bizkB7L4,1113
416
417
  lfx/components/models/embedding_model.py,sha256=hgfpY_3vc4l1v_qdCHQdJIyJ7UEUr4rbzyXzY0Cyec8,4212
417
418
  lfx/components/models/language_model.py,sha256=TA24DMAXrlruY3mNsfI9qGltfQ2h9cQpxe8DW8HLdeE,5992
@@ -438,7 +439,7 @@ lfx/components/openai/__init__.py,sha256=G4Fgw4pmmDohdIOmzaeSCGijzKjyqFXNJPLwlcU
438
439
  lfx/components/openai/openai.py,sha256=imWO1tTJ0tTLqax1v5bNBPCRINTj2f2wN8j5G-a07GI,4505
439
440
  lfx/components/openai/openai_chat_model.py,sha256=hZO79PqSI8ppnrEQFL_NL1issX6E5iCUPCycjX_d0Fs,6969
440
441
  lfx/components/openrouter/__init__.py,sha256=sWZTr41sX09AYpeyd2NWHQpx9jPuc1Zc4KdmErP08uM,950
441
- lfx/components/openrouter/openrouter.py,sha256=nNwKMAU44mK3ogidFfHjgsbBGbWK8EWQl39o82O0mzo,7341
442
+ lfx/components/openrouter/openrouter.py,sha256=g8PDEqeiP3C5iM5QDS_ABOeE1KNS_szBsX493u1dsic,3875
442
443
  lfx/components/output_parsers/__init__.py,sha256=aZ7X5AJUmAOVa9BZPSqQKALpF4ZFiZD-M2mad34zj-w,67
443
444
  lfx/components/perplexity/__init__.py,sha256=jITrDKNSr0CWAMIvoEaHSYkRG1OGy3zBOYIgblt5tNE,932
444
445
  lfx/components/perplexity/perplexity.py,sha256=uXCETjdu7XaIH9XJYhEeKOOGeDbTOA0JJElEbuxTQ6E,2764
@@ -528,30 +529,8 @@ lfx/components/upstash/upstash.py,sha256=tbLXjoxE3wFMawy_coPFQgmEVQvcTLolLuu3a0F
528
529
  lfx/components/vectara/__init__.py,sha256=sGf28Z5XdvL0fIxZzjoX6542fTb2Tiw0UesFz7Ii_lg,1065
529
530
  lfx/components/vectara/vectara.py,sha256=ojCeBGwGBGbpuSProTqz4MS74J-2THcZZ9uIpBTuRhE,3557
530
531
  lfx/components/vectara/vectara_rag.py,sha256=Ejye75GR37Wy9OdfCw7aJfjyPbbzaBxtTGoYZftzHDo,5674
531
- lfx/components/vectorstores/__init__.py,sha256=U8s0jKhFf-K4HULTiSp-Ssud94UMFArz-4nAh44jJy0,1194
532
- lfx/components/vectorstores/astradb.py,sha256=DibYwThNg1OBdk_fNh8Cv4NF8JoNf5K1fR4wREYsYus,53429
533
- lfx/components/vectorstores/astradb_graph.py,sha256=9HUnnjQqmQClYhgxtYR6bVuvhDimSXSVsS6bx-sMAKw,12663
534
- lfx/components/vectorstores/cassandra.py,sha256=gz8nfaq3ta6h_zVkHVn8uRKtcs9vDRuE7d1p5vI7LhE,9540
535
- lfx/components/vectorstores/cassandra_graph.py,sha256=JsY_gHRexI2p3U6uiQuygC7upu_HljquasDMId6PG6Y,8305
536
- lfx/components/vectorstores/chroma.py,sha256=DxvHJqhqx6CYdPoAG5Ly6jJsEOMR2R4lXv4UlvGQmQc,6171
537
- lfx/components/vectorstores/clickhouse.py,sha256=kQnoE9ycFPWDE0rNLxaTyjxq5b3aCeaf-f5H_Lx7iBs,5035
538
- lfx/components/vectorstores/couchbase.py,sha256=cO1HWm9IisWWVcxKjr7TuoOfqnY2Ndnq_QPP3L2yAoc,3965
539
- lfx/components/vectorstores/elasticsearch.py,sha256=WcBi8THcOzopZeYOQeEoHxsZkACHk4R3MKhSEYGxnfY,9773
540
- lfx/components/vectorstores/faiss.py,sha256=K9egZNckeHOrPqxuFia4VL4-mFphyEl6dQ_F-lXvax8,3893
541
- lfx/components/vectorstores/graph_rag.py,sha256=4NmYQkjSru_zaDhJfxdaYtap-RMGJfv2AYN2NEYSdds,5163
542
- lfx/components/vectorstores/hcd.py,sha256=Fo7Zj4U-A1ZcbsdhlTxheMJDy8EzbYWlo85iY6vASnQ,12379
532
+ lfx/components/vectorstores/__init__.py,sha256=GXlk9X_CfFFlRbwmUrCoT2Ey8vObZ_C1hS_04zZIx9Y,920
543
533
  lfx/components/vectorstores/local_db.py,sha256=bruzpEk6PQc1DeGqSWJblsVplJ5eC1JsSlncxRc4yuQ,10648
544
- lfx/components/vectorstores/milvus.py,sha256=_jMyO4l5y6O2Lkx9eHyuhgcHtXcApkI-ktuqa1YxnGI,4395
545
- lfx/components/vectorstores/mongodb_atlas.py,sha256=OlAstNMToHuvGI-8djkiGr7kdGBr927O0SE5cnVd0O0,8594
546
- lfx/components/vectorstores/opensearch.py,sha256=P8Eq4KsjHT8b7iOUOKMFRwOLwgRfIWfxIHLD0GJsw24,9080
547
- lfx/components/vectorstores/pgvector.py,sha256=UBF2B79eVfjj3hHoxrHzmT2UXOsUZxp4dWvyuPS2wh4,2635
548
- lfx/components/vectorstores/pinecone.py,sha256=VkygoOmrco417hYInjYIFwuxX1M7peYJl9-jhuiySR8,5137
549
- lfx/components/vectorstores/qdrant.py,sha256=1U2pdqXbo_HBgb8f8UUEkajfn9lq-SCUF_oiBuOE-40,4233
550
- lfx/components/vectorstores/supabase.py,sha256=UEW4HDQL_6GkPBsszowhrq8thouir--n2HE4DcAsD6k,2820
551
- lfx/components/vectorstores/upstash.py,sha256=bab5s-ALR4bT_pZhjFUlVHifrUachVFVQyxE6Hh5lgA,4083
552
- lfx/components/vectorstores/vectara.py,sha256=ojCeBGwGBGbpuSProTqz4MS74J-2THcZZ9uIpBTuRhE,3557
553
- lfx/components/vectorstores/vectara_rag.py,sha256=Ejye75GR37Wy9OdfCw7aJfjyPbbzaBxtTGoYZftzHDo,5674
554
- lfx/components/vectorstores/weaviate.py,sha256=ZrqR1qRA3IZsJB-TB07C7Qyqt5ysh-eEiHp7BcBtOsE,3280
555
534
  lfx/components/vertexai/__init__.py,sha256=bIcDPTzHyUujS75CWtFxYCvdY7eygvH3UDmWVAcX8I0,1090
556
535
  lfx/components/vertexai/vertexai.py,sha256=2fdSgdP6lfVYZElxWvuwi5wylpwFEtlSUlT1zaCGtgE,2992
557
536
  lfx/components/vertexai/vertexai_embeddings.py,sha256=Or1cFSZKKUp063yZVny7oqKiWgEqonhhCHPbRjY4CMA,3135
@@ -590,7 +569,7 @@ lfx/custom/code_parser/__init__.py,sha256=qIwZQdEp1z7ldn0z-GY44wmwRaywN3L6VPoPt6
590
569
  lfx/custom/code_parser/code_parser.py,sha256=QAqsp4QF607319dClK60BsaiwZLV55n0xeGR-DthSoE,14280
591
570
  lfx/custom/custom_component/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
592
571
  lfx/custom/custom_component/base_component.py,sha256=Pxi-qCocrGIwcG0x5fu-7ty1Py71bl_KG9Fku5SeO_M,4053
593
- lfx/custom/custom_component/component.py,sha256=6BIJxp6M8y1wCVUTGoQveYVDzeMDnuqcItcJpua_UUk,75795
572
+ lfx/custom/custom_component/component.py,sha256=rrtW0Ck0RtCIgyU7GfyY0IoR9Qk-70-7SzMdEXg5em4,75845
594
573
  lfx/custom/custom_component/component_with_cache.py,sha256=por6CiPL3EHdLp_DvfI7qz1n4tc1KkqMOJNbsxoqVaI,313
595
574
  lfx/custom/custom_component/custom_component.py,sha256=801FjGiQk7M7GD3CbU19AmCVS5KZjPVcKUy_wBXnm0o,22301
596
575
  lfx/custom/directory_reader/__init__.py,sha256=eFjlhKjpt2Kha_sJ2EqWofLRbpvfOTjvDSCpdpaTqWk,77
@@ -607,15 +586,15 @@ lfx/graph/__init__.py,sha256=uHjF2QW6i88_q3uspuPnulTyEA_QNV6eygOD8UZgG40,309
607
586
  lfx/graph/schema.py,sha256=58vzeUEseURXBjLFeEFFWcXmFSzeQczYdzIhQ_79Nlg,2668
608
587
  lfx/graph/utils.py,sha256=E0COGP-CzVNmIq-5hsqanTQw9T2bUtA_kEM53h3ZLYc,5920
609
588
  lfx/graph/edge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
610
- lfx/graph/edge/base.py,sha256=_FqhvevfgrJCzSxNdB2M2E3V6RGNPopQVo5FOxNbkis,12811
589
+ lfx/graph/edge/base.py,sha256=_5kYebmaNE-ZBVDa01WbsKHUMmlo2G2a9CoB1gEthPc,12867
611
590
  lfx/graph/edge/schema.py,sha256=bKlprxymeV04bTMw3jDLpYQAA3eRc4BIdD_-4XGGU3c,3806
612
591
  lfx/graph/edge/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
613
592
  lfx/graph/graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
614
593
  lfx/graph/graph/ascii.py,sha256=-jYWI_Zmz24mfLOxJrzIINZ0dbQgd9PXJfqwZa03xas,6211
615
- lfx/graph/graph/base.py,sha256=-USGP3n1TFH-OOncRktUJ2uY1FEXIfdLbcl2cJYM-2g,96048
594
+ lfx/graph/graph/base.py,sha256=H1Be8kSvHuA5qNlGCS_AeIW8k2VaCZ_3mEt9ak_fXAk,96047
616
595
  lfx/graph/graph/constants.py,sha256=jwjl4RydV_k_zawbI8FIgiLHeBBgH-cStVitxxSyXQs,1641
617
596
  lfx/graph/graph/runnable_vertices_manager.py,sha256=c-qQP3koKyAsIADDSONiiz4FIRIn6q5kAMX6EQIBBfA,6148
618
- lfx/graph/graph/schema.py,sha256=9uyjRqf18S6jLnLwSHhmWxO-XtpnAruugSjwqwp4ILE,1093
597
+ lfx/graph/graph/schema.py,sha256=1a8pictdRVN6ByqpvhOYu7heDeVs7A3eGpZuYIF7kaY,1086
619
598
  lfx/graph/graph/state_model.py,sha256=OJYIffeQ_3djtMdIsMmYduLr2JbX9GQ5N_SJ_PZeHJ4,2925
620
599
  lfx/graph/graph/utils.py,sha256=rsUaMjhpZzyIFD8QSf-G5J5PDTPFNJXxV5zDs5eLPqw,37745
621
600
  lfx/graph/state/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -627,7 +606,7 @@ lfx/graph/vertex/exceptions.py,sha256=QTe-7TRCI0TXswRZh1kh0Z3KySjQsJgY5zTU6o0jbo
627
606
  lfx/graph/vertex/param_handler.py,sha256=o6RmGvdo8rzasIZaalkAvC3LJA7oS8ETqpmXnOtarI8,12887
628
607
  lfx/graph/vertex/schema.py,sha256=3h6c7TTdZI1mzfAebhD-6CCCRLu1mQ8UDmgJijx5zWg,552
629
608
  lfx/graph/vertex/utils.py,sha256=iJmY4PXta5dYWTX2SMEbpfMKrzwkJVQmi8qstSv8D7I,738
630
- lfx/graph/vertex/vertex_types.py,sha256=r4eMlTUTWR5CQf-W34BBzpHpQ9KdWcK3L-2byE22enU,21140
609
+ lfx/graph/vertex/vertex_types.py,sha256=pHSzH0ePGWmYCbD67QlG3_kxRpM4OHbs3S6FYDj6Tjw,21145
631
610
  lfx/helpers/__init__.py,sha256=XUsPqOxQNoleWGC4EtVxqgaEQjR4cHsHOsyJ3Kjm88I,2912
632
611
  lfx/helpers/base_model.py,sha256=EiBdNJVE83BNKsg-IedyZYd78Mbl0m7BN2XTFeTlBhw,1956
633
612
  lfx/helpers/custom.py,sha256=9Z6rVfIrih27qsGkV1lzVpkK-ifpQuOaGSUog_w4asM,306
@@ -749,7 +728,7 @@ lfx/utils/schemas.py,sha256=NbOtVQBrn4d0BAu-0H_eCTZI2CXkKZlRY37XCSmuJwc,3865
749
728
  lfx/utils/util.py,sha256=Ww85wbr1-vjh2pXVtmTqoUVr6MXAW8S7eDx_Ys6HpE8,20696
750
729
  lfx/utils/util_strings.py,sha256=nU_IcdphNaj6bAPbjeL-c1cInQPfTBit8mp5Y57lwQk,1686
751
730
  lfx/utils/version.py,sha256=cHpbO0OJD2JQAvVaTH_6ibYeFbHJV0QDHs_YXXZ-bT8,671
752
- lfx_nightly-0.1.13.dev2.dist-info/METADATA,sha256=0f_0kz4XMwllPWDRJ-itUAZ0UyLud12fqO0m81UYhZM,8289
753
- lfx_nightly-0.1.13.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
754
- lfx_nightly-0.1.13.dev2.dist-info/entry_points.txt,sha256=1724p3RHDQRT2CKx_QRzEIa7sFuSVO0Ux70YfXfoMT4,42
755
- lfx_nightly-0.1.13.dev2.dist-info/RECORD,,
731
+ lfx_nightly-0.1.13.dev4.dist-info/METADATA,sha256=PhvMRiEpELV5wmolAxxBFyt3rp-Yz8P2YNGVToSQYR4,8289
732
+ lfx_nightly-0.1.13.dev4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
733
+ lfx_nightly-0.1.13.dev4.dist-info/entry_points.txt,sha256=1724p3RHDQRT2CKx_QRzEIa7sFuSVO0Ux70YfXfoMT4,42
734
+ lfx_nightly-0.1.13.dev4.dist-info/RECORD,,
@@ -1,77 +0,0 @@
1
- import os
2
-
3
- from lfx.base.memory.model import LCChatMemoryComponent
4
- from lfx.field_typing.constants import Memory
5
- from lfx.inputs.inputs import MessageTextInput, SecretStrInput, StrInput
6
-
7
-
8
- class AstraDBChatMemory(LCChatMemoryComponent):
9
- display_name = "Astra DB Chat Memory"
10
- description = "Retrieves and store chat messages from Astra DB."
11
- name = "AstraDBChatMemory"
12
- documentation: str = "https://docs.langflow.org/bundles-datastax#astra-db-chat-memory"
13
- icon: str = "AstraDB"
14
-
15
- inputs = [
16
- SecretStrInput(
17
- name="token",
18
- display_name="Astra DB Application Token",
19
- info="Authentication token for accessing Astra DB.",
20
- value="ASTRA_DB_APPLICATION_TOKEN",
21
- required=True,
22
- advanced=os.getenv("ASTRA_ENHANCED", "false").lower() == "true",
23
- ),
24
- SecretStrInput(
25
- name="api_endpoint",
26
- display_name="Astra DB API Endpoint",
27
- info="API endpoint URL for the Astra DB service.",
28
- value="ASTRA_DB_API_ENDPOINT",
29
- required=True,
30
- ),
31
- StrInput(
32
- name="collection_name",
33
- display_name="Collection Name",
34
- info="The name of the collection within Astra DB where the vectors will be stored.",
35
- required=True,
36
- ),
37
- StrInput(
38
- name="namespace",
39
- display_name="Namespace",
40
- info="Optional namespace within Astra DB to use for the collection.",
41
- advanced=True,
42
- ),
43
- MessageTextInput(
44
- name="session_id",
45
- display_name="Session ID",
46
- info="The session ID of the chat. If empty, the current session ID parameter will be used.",
47
- advanced=True,
48
- ),
49
- ]
50
-
51
- def build_message_history(self) -> Memory:
52
- try:
53
- from astrapy.admin import parse_api_endpoint
54
- from langchain_astradb.chat_message_histories import AstraDBChatMessageHistory
55
-
56
- except ImportError as e:
57
- msg = (
58
- "Could not import langchain Astra DB integration package. "
59
- "Please install it with `pip install langchain-astradb`."
60
- )
61
- raise ImportError(msg) from e
62
-
63
- try:
64
- from astrapy.admin import parse_api_endpoint
65
-
66
- except ImportError as e:
67
- msg = "Could not import astrapy package. "
68
- raise ImportError(msg) from e
69
-
70
- return AstraDBChatMessageHistory(
71
- session_id=self.session_id,
72
- collection_name=self.collection_name,
73
- token=self.token,
74
- api_endpoint=self.api_endpoint,
75
- namespace=self.namespace or None,
76
- environment=parse_api_endpoint(self.api_endpoint).environment,
77
- )
@@ -1,92 +0,0 @@
1
- from lfx.base.memory.model import LCChatMemoryComponent
2
- from lfx.field_typing.constants import Memory
3
- from lfx.inputs.inputs import DictInput, MessageTextInput, SecretStrInput
4
-
5
-
6
- class CassandraChatMemory(LCChatMemoryComponent):
7
- display_name = "Cassandra Chat Memory"
8
- description = "Retrieves and store chat messages from Apache Cassandra."
9
- name = "CassandraChatMemory"
10
- icon = "Cassandra"
11
-
12
- inputs = [
13
- MessageTextInput(
14
- name="database_ref",
15
- display_name="Contact Points / Astra Database ID",
16
- info="Contact points for the database (or Astra DB database ID)",
17
- required=True,
18
- ),
19
- MessageTextInput(
20
- name="username", display_name="Username", info="Username for the database (leave empty for Astra DB)."
21
- ),
22
- SecretStrInput(
23
- name="token",
24
- display_name="Password / Astra DB Token",
25
- info="User password for the database (or Astra DB token).",
26
- required=True,
27
- ),
28
- MessageTextInput(
29
- name="keyspace",
30
- display_name="Keyspace",
31
- info="Table Keyspace (or Astra DB namespace).",
32
- required=True,
33
- ),
34
- MessageTextInput(
35
- name="table_name",
36
- display_name="Table Name",
37
- info="The name of the table (or Astra DB collection) where vectors will be stored.",
38
- required=True,
39
- ),
40
- MessageTextInput(
41
- name="session_id", display_name="Session ID", info="Session ID for the message.", advanced=True
42
- ),
43
- DictInput(
44
- name="cluster_kwargs",
45
- display_name="Cluster arguments",
46
- info="Optional dictionary of additional keyword arguments for the Cassandra cluster.",
47
- advanced=True,
48
- is_list=True,
49
- ),
50
- ]
51
-
52
- def build_message_history(self) -> Memory:
53
- from langchain_community.chat_message_histories import CassandraChatMessageHistory
54
-
55
- try:
56
- import cassio
57
- except ImportError as e:
58
- msg = "Could not import cassio integration package. Please install it with `pip install cassio`."
59
- raise ImportError(msg) from e
60
-
61
- from uuid import UUID
62
-
63
- database_ref = self.database_ref
64
-
65
- try:
66
- UUID(self.database_ref)
67
- is_astra = True
68
- except ValueError:
69
- is_astra = False
70
- if "," in self.database_ref:
71
- # use a copy because we can't change the type of the parameter
72
- database_ref = self.database_ref.split(",")
73
-
74
- if is_astra:
75
- cassio.init(
76
- database_id=database_ref,
77
- token=self.token,
78
- cluster_kwargs=self.cluster_kwargs,
79
- )
80
- else:
81
- cassio.init(
82
- contact_points=database_ref,
83
- username=self.username,
84
- password=self.token,
85
- cluster_kwargs=self.cluster_kwargs,
86
- )
87
-
88
- return CassandraChatMessageHistory(
89
- session_id=self.session_id,
90
- table_name=self.table_name,
91
- keyspace=self.keyspace,
92
- )