mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-24 17:31:41 +00:00

* 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>
98 lines
2.3 KiB
TOML
98 lines
2.3 KiB
TOML
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[project]
|
|
name = "autogen-ext"
|
|
version = "0.4.1"
|
|
license = {file = "LICENSE-CODE"}
|
|
description = "AutoGen extensions library"
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
classifiers = [
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
]
|
|
dependencies = [
|
|
"autogen-core==0.4.1",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
langchain = ["langchain_core~= 0.3.3"]
|
|
azure = ["azure-core", "azure-identity"]
|
|
docker = ["docker~=7.0"]
|
|
openai = ["openai>=1.52.2", "tiktoken>=0.8.0", "aiofiles"]
|
|
file-surfer = [
|
|
"autogen-agentchat==0.4.1",
|
|
"markitdown>=0.0.1a2",
|
|
]
|
|
graphrag = ["graphrag>=1.0.1"]
|
|
web-surfer = [
|
|
"autogen-agentchat==0.4.1",
|
|
"playwright>=1.48.0",
|
|
"pillow>=11.0.0",
|
|
"markitdown>=0.0.1a2",
|
|
]
|
|
magentic-one = [
|
|
"autogen-agentchat==0.4.1",
|
|
"markitdown>=0.0.1a2",
|
|
"playwright>=1.48.0",
|
|
"pillow>=11.0.0",
|
|
]
|
|
video-surfer = [
|
|
"autogen-agentchat==0.4.1",
|
|
"opencv-python>=4.5",
|
|
"ffmpeg-python",
|
|
"openai-whisper",
|
|
]
|
|
|
|
grpc = [
|
|
"grpcio~=1.62.0", # TODO: update this once we have a stable version.
|
|
]
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
packages = ["src/autogen_ext"]
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"autogen_test_utils",
|
|
"langchain-experimental",
|
|
"pandas-stubs>=2.2.3.241126",
|
|
]
|
|
|
|
[tool.ruff]
|
|
extend = "../../pyproject.toml"
|
|
include = ["src/**", "tests/*.py"]
|
|
exclude = ["src/autogen_ext/agents/web_surfer/*.js", "src/autogen_ext/runtimes/grpc/protos", "tests/protos"]
|
|
|
|
[tool.pyright]
|
|
extends = "../../pyproject.toml"
|
|
include = ["src", "tests"]
|
|
exclude = ["src/autogen_ext/runtimes/grpc/protos", "tests/protos"]
|
|
|
|
[tool.pytest.ini_options]
|
|
minversion = "6.0"
|
|
testpaths = ["tests"]
|
|
|
|
[tool.poe]
|
|
include = "../../shared_tasks.toml"
|
|
|
|
[tool.poe.tasks]
|
|
test.sequence = [
|
|
"playwright install",
|
|
"pytest -n auto",
|
|
]
|
|
test.default_item_type = "cmd"
|
|
coverage.sequence = [
|
|
"playwright install",
|
|
"pytest -n auto --cov=src --cov-report=term-missing --cov-report=xml",
|
|
]
|
|
coverage.default_item_type = "cmd"
|
|
mypy = "mypy --config-file ../../pyproject.toml --exclude src/autogen_ext/runtimes/grpc/protos --exclude tests/protos src tests"
|
|
|
|
[tool.mypy]
|
|
[[tool.mypy.overrides]]
|
|
module = "docker.*"
|
|
ignore_missing_imports = true
|