datahub-agent-context 1.3.1.10rc1__py3-none-any.whl → 1.4.0rc2__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.
Files changed (42) hide show
  1. datahub_agent_context/__init__.py +11 -3
  2. datahub_agent_context/_version.py +1 -1
  3. datahub_agent_context/cli.py +152 -0
  4. datahub_agent_context/context.py +47 -34
  5. datahub_agent_context/langchain_tools/builder.py +6 -4
  6. datahub_agent_context/mcp_tools/base.py +6 -3
  7. datahub_agent_context/mcp_tools/save_document.py +634 -0
  8. datahub_agent_context/snowflake/__init__.py +0 -0
  9. datahub_agent_context/snowflake/generate_udfs.py +306 -0
  10. datahub_agent_context/snowflake/generators/__init__.py +21 -0
  11. datahub_agent_context/snowflake/generators/configuration.py +104 -0
  12. datahub_agent_context/snowflake/generators/cortex_agent.py +725 -0
  13. datahub_agent_context/snowflake/generators/network_rules.py +53 -0
  14. datahub_agent_context/snowflake/generators/stored_procedure.py +87 -0
  15. datahub_agent_context/snowflake/snowflake.py +662 -0
  16. datahub_agent_context/snowflake/udfs/__init__.py +1 -0
  17. datahub_agent_context/snowflake/udfs/add_glossary_terms.py +61 -0
  18. datahub_agent_context/snowflake/udfs/add_owners.py +59 -0
  19. datahub_agent_context/snowflake/udfs/add_structured_properties.py +57 -0
  20. datahub_agent_context/snowflake/udfs/add_tags.py +61 -0
  21. datahub_agent_context/snowflake/udfs/base.py +45 -0
  22. datahub_agent_context/snowflake/udfs/get_dataset_queries.py +68 -0
  23. datahub_agent_context/snowflake/udfs/get_entities.py +47 -0
  24. datahub_agent_context/snowflake/udfs/get_lineage.py +61 -0
  25. datahub_agent_context/snowflake/udfs/get_lineage_paths_between.py +69 -0
  26. datahub_agent_context/snowflake/udfs/get_me.py +51 -0
  27. datahub_agent_context/snowflake/udfs/grep_documents.py +70 -0
  28. datahub_agent_context/snowflake/udfs/list_schema_fields.py +80 -0
  29. datahub_agent_context/snowflake/udfs/remove_domains.py +45 -0
  30. datahub_agent_context/snowflake/udfs/remove_glossary_terms.py +57 -0
  31. datahub_agent_context/snowflake/udfs/remove_owners.py +56 -0
  32. datahub_agent_context/snowflake/udfs/remove_structured_properties.py +56 -0
  33. datahub_agent_context/snowflake/udfs/remove_tags.py +57 -0
  34. datahub_agent_context/snowflake/udfs/search_datahub.py +71 -0
  35. datahub_agent_context/snowflake/udfs/search_documents.py +58 -0
  36. datahub_agent_context/snowflake/udfs/set_domains.py +55 -0
  37. datahub_agent_context/snowflake/udfs/update_description.py +60 -0
  38. {datahub_agent_context-1.3.1.10rc1.dist-info → datahub_agent_context-1.4.0rc2.dist-info}/METADATA +21 -14
  39. datahub_agent_context-1.4.0rc2.dist-info/RECORD +66 -0
  40. datahub_agent_context-1.3.1.10rc1.dist-info/RECORD +0 -34
  41. {datahub_agent_context-1.3.1.10rc1.dist-info → datahub_agent_context-1.4.0rc2.dist-info}/WHEEL +0 -0
  42. {datahub_agent_context-1.3.1.10rc1.dist-info → datahub_agent_context-1.4.0rc2.dist-info}/top_level.txt +0 -0
datahub_agent_context-1.4.0rc2.dist-info/RECORD ADDED
@@ -0,0 +1,66 @@
1
+ datahub_agent_context/__init__.py,sha256=WgJFMZaA5ae_9ntP686UXd0TvZpbGwQRdSISi0JHsvU,967
2
+ datahub_agent_context/_version.py,sha256=sjXQOjG_dBO3UkwODDbFi7aLjoKMlkbubWQwtnl0qh0,648
3
+ datahub_agent_context/cli.py,sha256=ND0KLT3cFb6KnQl6kEb7B74tAOu6yfS4dO6mJjZW1x4,4441
4
+ datahub_agent_context/context.py,sha256=wj9q9hGf72q6oarnfEFHzqgS-vwtMO79hhjz8GNC0QQ,3163
5
+ datahub_agent_context/py.typed,sha256=kO13kg6OXApIRwKRcPpEOL09GZHx2Pk8Rp2KZpxv0lw,63
6
+ datahub_agent_context/langchain_tools/__init__.py,sha256=M0tn6fD9qY5Wc1XdptQuIf_7MSKLX8OSBaBxcPo5wmw,259
7
+ datahub_agent_context/langchain_tools/builder.py,sha256=-h8IuFWfIJGhQXnuK5ASCMwhoNmIhxLV2yu3ZOZPwpg,5288
8
+ datahub_agent_context/mcp_tools/__init__.py,sha256=7iUoWuT-KvszOqnmL3_co2LVQdhZtkQKRLRE98Hn8WM,1544
9
+ datahub_agent_context/mcp_tools/_token_estimator.py,sha256=U0kTqPZKBkKwxe7JZaLxIIFEobNSrEEHoM4NQbrmmAE,2782
10
+ datahub_agent_context/mcp_tools/base.py,sha256=UFqMe9yS-YikgaKOnu2DkaMLaHRzwSUBOFXOWxBjULA,11054
11
+ datahub_agent_context/mcp_tools/descriptions.py,sha256=xjfQJ1g-Vrj5L4EZ_Zd2_ZivI_utWnFsKkzV8Ad9aPA,10504
12
+ datahub_agent_context/mcp_tools/documents.py,sha256=en-W7pWs6C-h2u6cTMn6g-SY62BT6Hp8tz0GXcj7fDQ,16468
13
+ datahub_agent_context/mcp_tools/domains.py,sha256=mgSPJVlGhN_mdnvQsSJGqcluVKnUI72uMT067GzwU_w,7880
14
+ datahub_agent_context/mcp_tools/entities.py,sha256=H39aJn5BIUFEW9jJLnimdDmfpO1Ei0sx8FSeb4y3hFc,13408
15
+ datahub_agent_context/mcp_tools/get_me.py,sha256=1XUoov7slzAYfycq-NR8w8GxEKvgmBUOKoK2I2_0nNM,3111
16
+ datahub_agent_context/mcp_tools/helpers.py,sha256=NRIoVEB62vDWDg26UOFv-IhM8mEQd4hf8eL4iCJI18Y,17253
17
+ datahub_agent_context/mcp_tools/lineage.py,sha256=sJVR2jJkbGU_KjjtqZ8IJVOKDaIjDdtQKtAIxYWq71Q,26753
18
+ datahub_agent_context/mcp_tools/owners.py,sha256=LGZ5n5a3xRKSttay2NLf_rq97_Dl9pGIcVFi-l7uJK8,11798
19
+ datahub_agent_context/mcp_tools/queries.py,sha256=V4-yFcCi3c8r4Xy7XVKfQ7s3SsIWXMAHRrI8Sqf2g20,6864
20
+ datahub_agent_context/mcp_tools/save_document.py,sha256=r3nYUQSQjmgCybX7X39zu8Ef4FON284K_D1X9Y53cIE,23221
21
+ datahub_agent_context/mcp_tools/search.py,sha256=z5Hy1jLV4uDO26nb_oFuP5w6GX0DYcYWRIWn3kDp7dY,9880
22
+ datahub_agent_context/mcp_tools/structured_properties.py,sha256=amj7C-sbeAyctrXY_rpc2vCNTaJy2aTRx21TioeKEJk,15745
23
+ datahub_agent_context/mcp_tools/tags.py,sha256=5_Wg1Jqf_FgPgYuUV5bDwQ6J8t_sECcSM5yVtwQruPs,10814
24
+ datahub_agent_context/mcp_tools/terms.py,sha256=NHJ_PITAp0vUMij0o3-7Hd0a_tfM8TF59f4kayFcd0Q,11558
25
+ datahub_agent_context/mcp_tools/gql/__init__.py,sha256=KYWPHGi7sDM0DXrrXmhlR6_zhym1qNbtFhjYk1Ug6ss,579
26
+ datahub_agent_context/mcp_tools/gql/document_search.gql,sha256=velgPE43komCgAx5Tot8X8zV0ttoxZdTe5knqjMmS14,1634
27
+ datahub_agent_context/mcp_tools/gql/document_semantic_search.gql,sha256=lbckeTLY5idPgAhTBGZA-DIaR3lrSWAw_-imlAOgGIM,1606
28
+ datahub_agent_context/mcp_tools/gql/entity_details.gql,sha256=LEca6jvIQCib6c3ex0AZ8Wmxf1VHaPIVwvYkMLD0y1w,41169
29
+ datahub_agent_context/mcp_tools/gql/queries.gql,sha256=TIk_LHNSqJAFbwI9V354N-5Rx_pJyDP6JswcX63UMMc,624
30
+ datahub_agent_context/mcp_tools/gql/query_entity.gql,sha256=Nxm4lGz-hy95XoMPIyeWtK2QBRIaMEicZ2yAWgJAV2s,527
31
+ datahub_agent_context/mcp_tools/gql/read_documents.gql,sha256=sJRDgAjwjwplXJdVz-68ZwEVtsIZ6y1vYyw7oum7qu0,304
32
+ datahub_agent_context/mcp_tools/gql/search.gql,sha256=vIKrUmPbRY1mMsv7nNbEP337z6DbgZMQ1zXw_gd66zo,3457
33
+ datahub_agent_context/snowflake/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
34
+ datahub_agent_context/snowflake/generate_udfs.py,sha256=Det0VerrWm9KdUgR_UsdSE2W3HalmP4HKGZiYPRch3Q,11916
35
+ datahub_agent_context/snowflake/snowflake.py,sha256=N2ydOFn_FK-vJD9nUrr5nBhuDSCTwnsJk8ptK6rb0qE,21805
36
+ datahub_agent_context/snowflake/generators/__init__.py,sha256=8o-w8JBGDjnc52EHuWK88GzH0Qw1AXfcsO9Glb9M6Io,626
37
+ datahub_agent_context/snowflake/generators/configuration.py,sha256=Q3X953B4KbTg8KMBsWqUBo7TqPJuNrNJX2Lrgr26V9o,3870
38
+ datahub_agent_context/snowflake/generators/cortex_agent.py,sha256=ccD6E22nwx2-kdc8xs6gLApSvof1KluzwvrHuSgT9C0,25945
39
+ datahub_agent_context/snowflake/generators/network_rules.py,sha256=Gzh86tvQBLKsxqhzKjKXpqn9wB-2OM__WvLygIt4Z_c,2136
40
+ datahub_agent_context/snowflake/generators/stored_procedure.py,sha256=9YoLajphMgWuNqe-6m2gIVocDfUw3o8yDFk4wheX0-0,2579
41
+ datahub_agent_context/snowflake/udfs/__init__.py,sha256=wS5Auahmmo22oujVTAaxxFtgLEIymeP7UqaNheK8n6Q,56
42
+ datahub_agent_context/snowflake/udfs/add_glossary_terms.py,sha256=cPwMUbemAlp-gU8HB5mYOtRae10Rrnq1JUZ8M-eRs-o,2254
43
+ datahub_agent_context/snowflake/udfs/add_owners.py,sha256=upJAtpFGJ4RkfgtbYIMaefoH1dmN3NYzzevksZeKTzg,1997
44
+ datahub_agent_context/snowflake/udfs/add_structured_properties.py,sha256=mLpynz-354N6QgD1pek3KkvNPyc10Fn01k-fy2v4Pwo,2092
45
+ datahub_agent_context/snowflake/udfs/add_tags.py,sha256=b1jKjQikzkyMB42Fl3-4GTIs0IEdIE4KeS1ki9jQ5Y0,2182
46
+ datahub_agent_context/snowflake/udfs/base.py,sha256=Cv9Ryl7X2JwQqfyLGrQzW9yIMKevpveeC95YTKLKxg0,1493
47
+ datahub_agent_context/snowflake/udfs/get_dataset_queries.py,sha256=10L6VRAAlk21E78V3zhXa1_XlKhUQB_3bQB2c9vzVkk,2504
48
+ datahub_agent_context/snowflake/udfs/get_entities.py,sha256=JBhvbqIrrWtb_kXmf7bz8tKtG9nB4QtMkJwEaQsKDJs,1570
49
+ datahub_agent_context/snowflake/udfs/get_lineage.py,sha256=A7p9Ac2yw2wIghBhiUllwxFATEpcVTk4RnZXkvggtjQ,2170
50
+ datahub_agent_context/snowflake/udfs/get_lineage_paths_between.py,sha256=iyHbJNtNddFI5MtNQgfRiARli23QhI_QJ2w1w-g-0ao,2659
51
+ datahub_agent_context/snowflake/udfs/get_me.py,sha256=qPNwAoZ8Cbn6OIqG71n24hjqyfvVeVyDgND7PPSQG18,1467
52
+ datahub_agent_context/snowflake/udfs/grep_documents.py,sha256=wXUL8eYcwWhuTsUV42a4D_PAmXtzmPgfPvqU5l2rTpE,2632
53
+ datahub_agent_context/snowflake/udfs/list_schema_fields.py,sha256=Skb3GOW9hWLE1YitucVgHcVcN9N5pqWbk33jxwnkJYk,3064
54
+ datahub_agent_context/snowflake/udfs/remove_domains.py,sha256=cXg2D8EGXhD7L5IswGfatQ4XnmKGLq3pNXDa5gRtE6A,1427
55
+ datahub_agent_context/snowflake/udfs/remove_glossary_terms.py,sha256=mPGZmybl7EwF4n_wF7qtBLbXLJvig__YxSSteBVTntQ,2030
56
+ datahub_agent_context/snowflake/udfs/remove_owners.py,sha256=rcfJd8nbmdzraug4rV3ZiIuxTrQhZpvQXSoTrZ9NZco,1907
57
+ datahub_agent_context/snowflake/udfs/remove_structured_properties.py,sha256=N6vYgoaMxe2pJimvAUHld8caTRwdXMryD5tnqKS2qLw,1983
58
+ datahub_agent_context/snowflake/udfs/remove_tags.py,sha256=8OYbaCjzYcItApT0UEwIP3yaFXgGHS5exSUeo4tspFM,1941
59
+ datahub_agent_context/snowflake/udfs/search_datahub.py,sha256=-El0JnpkClaaxX5tCZxiTH7Nh9EIjJXB1_FpnnSd_Es,2631
60
+ datahub_agent_context/snowflake/udfs/search_documents.py,sha256=yrcLsSNyq_M-CsgwGrsgSXFVeusKAvuNqEI_DvHtC08,1944
61
+ datahub_agent_context/snowflake/udfs/set_domains.py,sha256=nKFJ9ZMBCz9qWgJv-FkOkeEmRE-QYBUCKbHw1RFdJmc,1701
62
+ datahub_agent_context/snowflake/udfs/update_description.py,sha256=laJjqqRARDy3VCo0xX_5lvTrZn2U3Dwm7XEliPdDRSA,2096
63
+ datahub_agent_context-1.4.0rc2.dist-info/METADATA,sha256=GzhPDkJ4HtHmsyCeu_Lm_z44amlbDRHevW7R9hSH0Y8,7795
64
+ datahub_agent_context-1.4.0rc2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
65
+ datahub_agent_context-1.4.0rc2.dist-info/top_level.txt,sha256=Tv1bg7ZwDOKM9u9RHj5m1Zbx2LDf4lVBBRNHi_gBBTI,22
66
+ datahub_agent_context-1.4.0rc2.dist-info/RECORD,,
datahub_agent_context-1.3.1.10rc1.dist-info/RECORD DELETED
@@ -1,34 +0,0 @@
1
- datahub_agent_context/__init__.py,sha256=VGBOuNztxuwUi5Ofnrpe7tw8EmUrQD-i-eMbSKwvMtU,890
2
- datahub_agent_context/_version.py,sha256=UkZ3xD-ir97HRLYqIDQIdvJvbuniEJPAH6GAJIlG4ss,651
3
- datahub_agent_context/context.py,sha256=qRc44o38Y-LoDQH1oFm38hIatOWRKnRxxytQNIR93kU,2771
4
- datahub_agent_context/py.typed,sha256=kO13kg6OXApIRwKRcPpEOL09GZHx2Pk8Rp2KZpxv0lw,63
5
- datahub_agent_context/langchain_tools/__init__.py,sha256=M0tn6fD9qY5Wc1XdptQuIf_7MSKLX8OSBaBxcPo5wmw,259
6
- datahub_agent_context/langchain_tools/builder.py,sha256=X59zdmdUqltKiTo3HZrE4-JOd7CztGprW2O32jIYt2o,5145
7
- datahub_agent_context/mcp_tools/__init__.py,sha256=7iUoWuT-KvszOqnmL3_co2LVQdhZtkQKRLRE98Hn8WM,1544
8
- datahub_agent_context/mcp_tools/_token_estimator.py,sha256=U0kTqPZKBkKwxe7JZaLxIIFEobNSrEEHoM4NQbrmmAE,2782
9
- datahub_agent_context/mcp_tools/base.py,sha256=r0vHc6uivWjTyGdpAkcmyvg4XuBTKWDds1Uv5BC13y4,10989
10
- datahub_agent_context/mcp_tools/descriptions.py,sha256=xjfQJ1g-Vrj5L4EZ_Zd2_ZivI_utWnFsKkzV8Ad9aPA,10504
11
- datahub_agent_context/mcp_tools/documents.py,sha256=en-W7pWs6C-h2u6cTMn6g-SY62BT6Hp8tz0GXcj7fDQ,16468
12
- datahub_agent_context/mcp_tools/domains.py,sha256=mgSPJVlGhN_mdnvQsSJGqcluVKnUI72uMT067GzwU_w,7880
13
- datahub_agent_context/mcp_tools/entities.py,sha256=H39aJn5BIUFEW9jJLnimdDmfpO1Ei0sx8FSeb4y3hFc,13408
14
- datahub_agent_context/mcp_tools/get_me.py,sha256=1XUoov7slzAYfycq-NR8w8GxEKvgmBUOKoK2I2_0nNM,3111
15
- datahub_agent_context/mcp_tools/helpers.py,sha256=NRIoVEB62vDWDg26UOFv-IhM8mEQd4hf8eL4iCJI18Y,17253
16
- datahub_agent_context/mcp_tools/lineage.py,sha256=sJVR2jJkbGU_KjjtqZ8IJVOKDaIjDdtQKtAIxYWq71Q,26753
17
- datahub_agent_context/mcp_tools/owners.py,sha256=LGZ5n5a3xRKSttay2NLf_rq97_Dl9pGIcVFi-l7uJK8,11798
18
- datahub_agent_context/mcp_tools/queries.py,sha256=V4-yFcCi3c8r4Xy7XVKfQ7s3SsIWXMAHRrI8Sqf2g20,6864
19
- datahub_agent_context/mcp_tools/search.py,sha256=z5Hy1jLV4uDO26nb_oFuP5w6GX0DYcYWRIWn3kDp7dY,9880
20
- datahub_agent_context/mcp_tools/structured_properties.py,sha256=amj7C-sbeAyctrXY_rpc2vCNTaJy2aTRx21TioeKEJk,15745
21
- datahub_agent_context/mcp_tools/tags.py,sha256=5_Wg1Jqf_FgPgYuUV5bDwQ6J8t_sECcSM5yVtwQruPs,10814
22
- datahub_agent_context/mcp_tools/terms.py,sha256=NHJ_PITAp0vUMij0o3-7Hd0a_tfM8TF59f4kayFcd0Q,11558
23
- datahub_agent_context/mcp_tools/gql/__init__.py,sha256=KYWPHGi7sDM0DXrrXmhlR6_zhym1qNbtFhjYk1Ug6ss,579
24
- datahub_agent_context/mcp_tools/gql/document_search.gql,sha256=velgPE43komCgAx5Tot8X8zV0ttoxZdTe5knqjMmS14,1634
25
- datahub_agent_context/mcp_tools/gql/document_semantic_search.gql,sha256=lbckeTLY5idPgAhTBGZA-DIaR3lrSWAw_-imlAOgGIM,1606
26
- datahub_agent_context/mcp_tools/gql/entity_details.gql,sha256=LEca6jvIQCib6c3ex0AZ8Wmxf1VHaPIVwvYkMLD0y1w,41169
27
- datahub_agent_context/mcp_tools/gql/queries.gql,sha256=TIk_LHNSqJAFbwI9V354N-5Rx_pJyDP6JswcX63UMMc,624
28
- datahub_agent_context/mcp_tools/gql/query_entity.gql,sha256=Nxm4lGz-hy95XoMPIyeWtK2QBRIaMEicZ2yAWgJAV2s,527
29
- datahub_agent_context/mcp_tools/gql/read_documents.gql,sha256=sJRDgAjwjwplXJdVz-68ZwEVtsIZ6y1vYyw7oum7qu0,304
30
- datahub_agent_context/mcp_tools/gql/search.gql,sha256=vIKrUmPbRY1mMsv7nNbEP337z6DbgZMQ1zXw_gd66zo,3457
31
- datahub_agent_context-1.3.1.10rc1.dist-info/METADATA,sha256=eeLSMxbmfo6ktfCQ4fjsV7yR69TGVr8ZS1bqQtXV-70,7418
32
- datahub_agent_context-1.3.1.10rc1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
33
- datahub_agent_context-1.3.1.10rc1.dist-info/top_level.txt,sha256=Tv1bg7ZwDOKM9u9RHj5m1Zbx2LDf4lVBBRNHi_gBBTI,22
34
- datahub_agent_context-1.3.1.10rc1.dist-info/RECORD,,