mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-28 06:33:15 +08:00
60a6c20174
* support mcp # Conflicts: # tensorrt_llm/scaffolding/worker.py Signed-off-by: wu1du2 <wu1du2@gmail.com> * move all into contrib/mcp # Conflicts: # examples/scaffolding/contrib/mcp/mcptest.py # tensorrt_llm/scaffolding/__init__.py # tensorrt_llm/scaffolding/contrib/__init__.py # tensorrt_llm/scaffolding/contrib/mcp/__init__.py # tensorrt_llm/scaffolding/contrib/mcp/mcp_controller.py # tensorrt_llm/scaffolding/task.py # tensorrt_llm/scaffolding/worker.py Signed-off-by: wu1du2 <wu1du2@gmail.com> * support sandbox, websearch # Conflicts: # examples/scaffolding/contrib/mcp/mcptest.py # examples/scaffolding/contrib/mcp/weather/weather.py # tensorrt_llm/scaffolding/contrib/mcp/mcp_controller.py # tensorrt_llm/scaffolding/contrib/mcp/mcp_utils.py # tensorrt_llm/scaffolding/contrib/mcp/mcp_worker.py # tensorrt_llm/scaffolding/worker.py Signed-off-by: wu1du2 <wu1du2@gmail.com> * remove pics Signed-off-by: wu1du2 <wu1du2@gmail.com> * pre-commit fix # Conflicts: # tensorrt_llm/scaffolding/contrib/mcp/__init__.py # tensorrt_llm/scaffolding/contrib/mcp/mcp_utils.py # tensorrt_llm/scaffolding/contrib/mcp/mcp_worker.py Signed-off-by: wu1du2 <wu1du2@gmail.com> * fix spell Signed-off-by: wu1du2 <wu1du2@gmail.com> * rebase Signed-off-by: wu1du2 <wu1du2@gmail.com> --------- Signed-off-by: wu1du2 <wu1du2@gmail.com>
22 lines
553 B
Python
22 lines
553 B
Python
from tensorrt_llm.scaffolding import * # noqa
|
|
|
|
from .AsyncGeneration import StreamGenerationTask, stream_generation_handler
|
|
from .Dynasor import DynasorGenerationController
|
|
from .mcp import (ChatTask, MCPCallTask, MCPController, MCPListTask, MCPWorker,
|
|
chat_handler)
|
|
|
|
__all__ = [
|
|
# AsyncGeneration
|
|
"stream_generation_handler",
|
|
"StreamGenerationTask",
|
|
# Dynasor
|
|
"DynasorGenerationController",
|
|
#mcp
|
|
"MCPController",
|
|
"MCPWorker",
|
|
"MCPCallTask",
|
|
"MCPListTask",
|
|
"ChatTask",
|
|
"chat_handler"
|
|
]
|