mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
90763daf81
Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Laure Hugo <201583486+laure0303@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
|
|
from vibe import __version__
|
|
from vibe.core.agent_loop import AgentLoop, TeleportError
|
|
from vibe.core.agents.models import BuiltinAgentName
|
|
from vibe.core.config import VibeConfig
|
|
from vibe.core.logger import logger
|
|
from vibe.core.output_formatters import create_formatter
|
|
from vibe.core.teleport.types import (
|
|
TeleportPushRequiredEvent,
|
|
TeleportPushResponseEvent,
|
|
)
|
|
from vibe.core.types import (
|
|
AssistantEvent,
|
|
ClientMetadata,
|
|
EntrypointMetadata,
|
|
LLMMessage,
|
|
OutputFormat,
|
|
Role,
|
|
)
|
|
from vibe.core.utils import ConversationLimitException
|
|
|
|
__all__ = ["TeleportError", "run_programmatic"]
|
|
|
|
_DEFAULT_CLIENT_METADATA = ClientMetadata(name="vibe_programmatic", version=__version__)
|
|
|
|
|
|
def run_programmatic(
|
|
config: VibeConfig,
|
|
prompt: str,
|
|
max_turns: int | None = None,
|
|
max_price: float | None = None,
|
|
output_format: OutputFormat = OutputFormat.TEXT,
|
|
previous_messages: list[LLMMessage] | None = None,
|
|
agent_name: str = BuiltinAgentName.AUTO_APPROVE,
|
|
client_metadata: ClientMetadata = _DEFAULT_CLIENT_METADATA,
|
|
teleport: bool = False,
|
|
) -> str | None:
|
|
formatter = create_formatter(output_format)
|
|
|
|
agent_loop = AgentLoop(
|
|
config,
|
|
agent_name=agent_name,
|
|
message_observer=formatter.on_message_added,
|
|
max_turns=max_turns,
|
|
max_price=max_price,
|
|
enable_streaming=False,
|
|
entrypoint_metadata=EntrypointMetadata(
|
|
agent_entrypoint="programmatic",
|
|
agent_version=__version__,
|
|
client_name=client_metadata.name,
|
|
client_version=client_metadata.version,
|
|
),
|
|
)
|
|
logger.info("USER: %s", prompt)
|
|
|
|
async def _async_run() -> str | None:
|
|
try:
|
|
if previous_messages:
|
|
non_system_messages = [
|
|
msg for msg in previous_messages if not (msg.role == Role.system)
|
|
]
|
|
agent_loop.messages.extend(non_system_messages)
|
|
logger.info(
|
|
"Loaded %d messages from previous session", len(non_system_messages)
|
|
)
|
|
|
|
agent_loop.emit_new_session_telemetry()
|
|
|
|
if teleport and config.nuage_enabled:
|
|
gen = agent_loop.teleport_to_vibe_nuage(prompt or None)
|
|
async for event in gen:
|
|
formatter.on_event(event)
|
|
if isinstance(event, TeleportPushRequiredEvent):
|
|
next_event = await gen.asend(
|
|
TeleportPushResponseEvent(approved=True)
|
|
)
|
|
formatter.on_event(next_event)
|
|
else:
|
|
async for event in agent_loop.act(prompt):
|
|
formatter.on_event(event)
|
|
if (
|
|
isinstance(event, AssistantEvent)
|
|
and event.stopped_by_middleware
|
|
):
|
|
raise ConversationLimitException(event.content)
|
|
|
|
return formatter.finalize()
|
|
finally:
|
|
await agent_loop.telemetry_client.aclose()
|
|
|
|
return asyncio.run(_async_run())
|