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.

model_config_template.json 1.3 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
1234567891011121314151617181920212223242526272829303132333435363738
  1. // Use Azure OpenAI with AD token provider.
  2. // {
  3. // "provider": "AzureOpenAIChatCompletionClient",
  4. // "config": {
  5. // "model": "gpt-4o-2024-05-13",
  6. // "azure_endpoint": "https://{your-custom-endpoint}.openai.azure.com/",
  7. // "azure_deployment": "{your-azure-deployment}",
  8. // "api_version": "2024-06-01",
  9. // "azure_ad_token_provider": {
  10. // "provider": "autogen_ext.auth.azure.AzureTokenProvider",
  11. // "config": {
  12. // "provider_kind": "DefaultAzureCredential",
  13. // "scopes": [
  14. // "https://cognitiveservices.azure.com/.default"
  15. // ]
  16. // }
  17. // }
  18. // }
  19. // }
  20. // Use Azure Open AI with key
  21. // {
  22. // "provider": "AzureOpenAIChatCompletionClient",
  23. // "config": {
  24. // "model": "gpt-4o-2024-05-13",
  25. // "azure_endpoint": "https://{your-custom-endpoint}.openai.azure.com/",
  26. // "azure_deployment": "{your-azure-deployment}",
  27. // "api_version": "2024-06-01",
  28. // "api_key": "REPLACE_WITH_YOUR_API_KEY"
  29. // }
  30. // }
  31. // Use Open AI with key
  32. {
  33. "provider": "OpenAIChatCompletionClient",
  34. "config": {
  35. "model": "gpt-4o-2024-05-13",
  36. "api_key": "REPLACE_WITH_YOUR_API_KEY"
  37. }
  38. }