phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 09:21:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

agents remove imports

Browse source
This commit is contained in:
Xi Yan 2024-12-26 18:47:46 -08:00
parent b936503784
commit 50764d76a7
1 changed files with 43 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

50
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -13,19 +13,55 @@ import secrets
import string import string
import uuid import uuid
from datetime import datetime from datetime import datetime
from typing import AsyncGenerator, List, Tuple from typing import AsyncGenerator, Dict, List, Tuple
from urllib.parse import urlparse from urllib.parse import urlparse
import httpx import httpx
from llama_models.llama3.api.datatypes import BuiltinTool
from llama_stack.apis.agents import * # noqa: F403 from llama_stack.apis.agents import (
from llama_stack.apis.inference import * # noqa: F403 AgentConfig,
from llama_stack.apis.memory import * # noqa: F403 AgentTool,
from llama_stack.apis.memory_banks import * # noqa: F403 AgentTurnCreateRequest,
from llama_stack.apis.safety import * # noqa: F403 AgentTurnResponseEvent,
AgentTurnResponseEventType,
AgentTurnResponseStepCompletePayload,
AgentTurnResponseStepStartPayload,
AgentTurnResponseStreamChunk,
AgentTurnResponseTurnCompletePayload,
AgentTurnResponseTurnStartPayload,
Attachment,
CodeInterpreterToolDefinition,
FunctionCallToolDefinition,
MemoryRetrievalStep,
PhotogenToolDefinition,
SearchToolDefinition,
ShieldCallStep,
StepType,
Turn,
WolframAlphaToolDefinition,
)
from llama_stack.apis.common.content_types import InterleavedContent, TextContentItem from llama_stack.apis.common.content_types import (
InterleavedContent,
TextContentItem,
URL,
)
from llama_stack.apis.inference import (
CompletionMessage,
Inference,
Message,
SamplingParams,
StopReason,
SystemMessage,
ToolDefinition,
ToolResponseMessage,
UserMessage,
)
from llama_stack.apis.memory import Memory, MemoryBankDocument
from llama_stack.apis.memory_banks import MemoryBanks
from llama_stack.apis.safety import Safety
from llama_stack.providers.utils.kvstore import KVStore from llama_stack.providers.utils.kvstore import KVStore
from llama_stack.providers.utils.memory.vector_store import concat_interleaved_content from llama_stack.providers.utils.memory.vector_store import concat_interleaved_content