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.5 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
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172
  1. [build-system]
  2. requires = ["hatchling"]
  3. build-backend = "hatchling.build"
  4. [project]
  5. name = "autogen-ext"
  6. version = "0.4.8"
  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.8",
  18. ]
  19. [project.optional-dependencies]
  20. langchain = ["langchain_core~= 0.3.3"]
  21. azure = [
  22. "azure-ai-inference>=1.0.0b7",
  23. "azure-core",
  24. "azure-identity",
  25. ]
  26. docker = ["docker~=7.0", "asyncio_atexit>=1.0.1"]
  27. openai = ["openai>=1.52.2", "tiktoken>=0.8.0", "aiofiles"]
  28. file-surfer = [
  29. "autogen-agentchat==0.4.8",
  30. "markitdown>=0.0.1a2",
  31. ]
  32. graphrag = ["graphrag>=1.0.1"]
  33. web-surfer = [
  34. "autogen-agentchat==0.4.8",
  35. "playwright>=1.48.0",
  36. "pillow>=11.0.0",
  37. "markitdown>=0.0.1a2",
  38. ]
  39. magentic-one = [
  40. "autogen-agentchat==0.4.8",
  41. "markitdown>=0.0.1a2",
  42. "playwright>=1.48.0",
  43. "pillow>=11.0.0",
  44. ]
  45. video-surfer = [
  46. "autogen-agentchat==0.4.8",
  47. "opencv-python>=4.5",
  48. "ffmpeg-python",
  49. "openai-whisper",
  50. ]
  51. diskcache = [
  52. "diskcache>=5.6.3"
  53. ]
  54. redis = [
  55. "redis>=5.2.1"
  56. ]
  57. grpc = [
  58. "grpcio~=1.70.0",
  59. ]
  60. jupyter-executor = [
  61. "ipykernel>=6.29.5",
  62. "nbclient>=0.10.2",
  63. ]
  64. semantic-kernel-core = [
  65. "semantic-kernel>=1.17.1",
  66. ]
  67. gemini = [
  68. "google-genai>=1.0.0",
  69. ]
  70. semantic-kernel-google = [
  71. "semantic-kernel[google]>=1.17.1",
  72. ]
  73. semantic-kernel-hugging-face = [
  74. "semantic-kernel[hugging_face]>=1.17.1",
  75. ]
  76. semantic-kernel-mistralai = [
  77. "semantic-kernel[mistralai]>=1.17.1",
  78. ]
  79. semantic-kernel-ollama = [
  80. "semantic-kernel[ollama]>=1.17.1",
  81. ]
  82. semantic-kernel-onnx = [
  83. "semantic-kernel[onnx]>=1.17.1",
  84. ]
  85. semantic-kernel-anthropic = [
  86. "semantic-kernel[anthropic]>=1.17.1",
  87. ]
  88. semantic-kernel-pandas = [
  89. "semantic-kernel[pandas]>=1.17.1",
  90. ]
  91. semantic-kernel-aws = [
  92. "semantic-kernel[aws]>=1.17.1",
  93. ]
  94. semantic-kernel-dapr = [
  95. "semantic-kernel[dapr]>=1.17.1",
  96. ]
  97. http-tool = [
  98. "httpx>=0.27.0",
  99. "json-schema-to-pydantic>=0.2.0"
  100. ]
  101. semantic-kernel-all = [
  102. "semantic-kernel[google,hugging_face,mistralai,ollama,onnx,anthropic,usearch,pandas,aws,dapr]>=1.17.1",
  103. ]
  104. rich = ["rich>=13.9.4"]
  105. mcp = [
  106. "mcp>=1.1.3",
  107. "json-schema-to-pydantic>=0.2.2"
  108. ]
  109. [tool.hatch.build.targets.wheel]
  110. packages = ["src/autogen_ext"]
  111. [dependency-groups]
  112. dev = [
  113. "autogen_test_utils",
  114. "langchain-experimental",
  115. "pandas-stubs>=2.2.3.241126",
  116. "httpx>=0.28.1",
  117. ]
  118. [tool.ruff]
  119. extend = "../../pyproject.toml"
  120. include = ["src/**", "tests/*.py"]
  121. exclude = ["src/autogen_ext/agents/web_surfer/*.js", "src/autogen_ext/runtimes/grpc/protos", "tests/protos"]
  122. [tool.pyright]
  123. extends = "../../pyproject.toml"
  124. include = ["src", "tests"]
  125. exclude = ["src/autogen_ext/runtimes/grpc/protos", "tests/protos"]
  126. [tool.pytest.ini_options]
  127. minversion = "6.0"
  128. testpaths = ["tests"]
  129. markers = [
  130. "grpc",
  131. ]
  132. [tool.poe]
  133. include = "../../shared_tasks.toml"
  134. [tool.poe.tasks]
  135. test.sequence = [
  136. "playwright install",
  137. "pytest -n 1 --cov=src --cov-report=term-missing --cov-report=xml",
  138. ]
  139. test.default_item_type = "cmd"
  140. test-grpc = "pytest -n 1 --cov=src --cov-report=term-missing --cov-report=xml --grpc"
  141. mypy = "mypy --config-file ../../pyproject.toml --exclude src/autogen_ext/runtimes/grpc/protos --exclude tests/protos src tests"
  142. [tool.mypy]
  143. [[tool.mypy.overrides]]
  144. module = "docker.*"
  145. ignore_missing_imports = true