You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

pyproject.toml 3.2 kB

Graphrag integration (#4612) * add initial global search draft * add graphrag dep * fix local search embedding * linting * add from config constructor * remove draft notebook * update config factory and add docstrings * add graphrag sample * add sample prompts * update readme * update deps * Add API docs * Update python/samples/agentchat_graphrag/requirements.txt * Update python/samples/agentchat_graphrag/requirements.txt * update docstrings with snippet and doc ref * lint * improve set up instructions in docstring * lint * update lock * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_global_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_local_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * add unit tests * update lock * update uv lock * add docstring newlines * stubs and typing on graphrag tests * fix docstrings * fix mypy error * + linting and type fixes * type fix graphrag sample * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_global_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_local_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * Update python/samples/agentchat_graphrag/requirements.txt Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * update overrides * fix docstring client imports * additional docstring fix * add docstring missing import * use openai and fix db path * use console for displaying messages * add model config and gitignore * update readme * lint * Update python/samples/agentchat_graphrag/README.md * Update python/samples/agentchat_graphrag/README.md * Comment remaining azure config --------- Co-authored-by: Leonardo Pinheiro <lpinheiro@microsoft.com> Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
1 year ago
Graphrag integration (#4612) * add initial global search draft * add graphrag dep * fix local search embedding * linting * add from config constructor * remove draft notebook * update config factory and add docstrings * add graphrag sample * add sample prompts * update readme * update deps * Add API docs * Update python/samples/agentchat_graphrag/requirements.txt * Update python/samples/agentchat_graphrag/requirements.txt * update docstrings with snippet and doc ref * lint * improve set up instructions in docstring * lint * update lock * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_global_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_local_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * add unit tests * update lock * update uv lock * add docstring newlines * stubs and typing on graphrag tests * fix docstrings * fix mypy error * + linting and type fixes * type fix graphrag sample * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_global_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * Update python/packages/autogen-ext/src/autogen_ext/tools/graphrag/_local_search.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * Update python/samples/agentchat_graphrag/requirements.txt Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * update overrides * fix docstring client imports * additional docstring fix * add docstring missing import * use openai and fix db path * use console for displaying messages * add model config and gitignore * update readme * lint * Update python/samples/agentchat_graphrag/README.md * Update python/samples/agentchat_graphrag/README.md * Comment remaining azure config --------- Co-authored-by: Leonardo Pinheiro <lpinheiro@microsoft.com> Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
1 year ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. [build-system]
  2. requires = ["hatchling"]
  3. build-backend = "hatchling.build"
  4. [project]
  5. name = "autogen-ext"
  6. version = "0.4.3"
  7. license = {file = "LICENSE-CODE"}
  8. description = "AutoGen extensions library"
  9. readme = "README.md"
  10. requires-python = ">=3.10"
  11. classifiers = [
  12. "Programming Language :: Python :: 3",
  13. "License :: OSI Approved :: MIT License",
  14. "Operating System :: OS Independent",
  15. ]
  16. dependencies = [
  17. "autogen-core==0.4.3",
  18. ]
  19. [project.optional-dependencies]
  20. langchain = ["langchain_core~= 0.3.3"]
  21. azure = ["azure-core", "azure-identity"]
  22. docker = ["docker~=7.0"]
  23. openai = ["openai>=1.52.2", "tiktoken>=0.8.0", "aiofiles"]
  24. file-surfer = [
  25. "autogen-agentchat==0.4.3",
  26. "markitdown>=0.0.1a2",
  27. ]
  28. graphrag = ["graphrag>=1.0.1"]
  29. web-surfer = [
  30. "autogen-agentchat==0.4.3",
  31. "playwright>=1.48.0",
  32. "pillow>=11.0.0",
  33. "markitdown>=0.0.1a2",
  34. ]
  35. magentic-one = [
  36. "autogen-agentchat==0.4.3",
  37. "markitdown>=0.0.1a2",
  38. "playwright>=1.48.0",
  39. "pillow>=11.0.0",
  40. ]
  41. video-surfer = [
  42. "autogen-agentchat==0.4.3",
  43. "opencv-python>=4.5",
  44. "ffmpeg-python",
  45. "openai-whisper",
  46. ]
  47. diskcache = [
  48. "diskcache>=5.6.3"
  49. ]
  50. redis = [
  51. "redis>=5.2.1"
  52. ]
  53. grpc = [
  54. "grpcio~=1.62.0", # TODO: update this once we have a stable version.
  55. ]
  56. jupyter-executor = [
  57. "ipykernel>=6.29.5",
  58. "nbclient>=0.10.2",
  59. ]
  60. semantic-kernel-core = [
  61. "semantic-kernel>=1.17.1",
  62. ]
  63. semantic-kernel-google = [
  64. "semantic-kernel[google]>=1.17.1",
  65. ]
  66. semantic-kernel-hugging-face = [
  67. "semantic-kernel[hugging_face]>=1.17.1",
  68. ]
  69. semantic-kernel-mistralai = [
  70. "semantic-kernel[mistralai]>=1.17.1",
  71. ]
  72. semantic-kernel-ollama = [
  73. "semantic-kernel[ollama]>=1.17.1",
  74. ]
  75. semantic-kernel-onnx = [
  76. "semantic-kernel[onnx]>=1.17.1",
  77. ]
  78. semantic-kernel-anthropic = [
  79. "semantic-kernel[anthropic]>=1.17.1",
  80. ]
  81. semantic-kernel-pandas = [
  82. "semantic-kernel[pandas]>=1.17.1",
  83. ]
  84. semantic-kernel-aws = [
  85. "semantic-kernel[aws]>=1.17.1",
  86. ]
  87. semantic-kernel-dapr = [
  88. "semantic-kernel[dapr]>=1.17.1",
  89. ]
  90. semantic-kernel-all = [
  91. "semantic-kernel[google,hugging_face,mistralai,ollama,onnx,anthropic,usearch,pandas,aws,dapr]>=1.17.1",
  92. ]
  93. [tool.hatch.build.targets.wheel]
  94. packages = ["src/autogen_ext"]
  95. [dependency-groups]
  96. dev = [
  97. "autogen_test_utils",
  98. "langchain-experimental",
  99. "pandas-stubs>=2.2.3.241126",
  100. ]
  101. [tool.ruff]
  102. extend = "../../pyproject.toml"
  103. include = ["src/**", "tests/*.py"]
  104. exclude = ["src/autogen_ext/agents/web_surfer/*.js", "src/autogen_ext/runtimes/grpc/protos", "tests/protos"]
  105. [tool.pyright]
  106. extends = "../../pyproject.toml"
  107. include = ["src", "tests"]
  108. exclude = ["src/autogen_ext/runtimes/grpc/protos", "tests/protos"]
  109. [tool.pytest.ini_options]
  110. minversion = "6.0"
  111. testpaths = ["tests"]
  112. [tool.poe]
  113. include = "../../shared_tasks.toml"
  114. [tool.poe.tasks]
  115. test.sequence = [
  116. "playwright install",
  117. "pytest -n 1 --cov=src --cov-report=term-missing --cov-report=xml",
  118. ]
  119. test.default_item_type = "cmd"
  120. mypy = "mypy --config-file ../../pyproject.toml --exclude src/autogen_ext/runtimes/grpc/protos --exclude tests/protos src tests"
  121. [tool.mypy]
  122. [[tool.mypy.overrides]]
  123. module = "docker.*"
  124. ignore_missing_imports = true