mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-03 12:12:15 +00:00
improve
This commit is contained in:
parent
c7a5795ab2
commit
02280d3b6b
2 changed files with 0 additions and 5 deletions
|
|
@ -15,10 +15,6 @@ class TGIImplConfig(BaseModel):
|
||||||
url: str = Field(
|
url: str = Field(
|
||||||
description="The URL for the TGI serving endpoint",
|
description="The URL for the TGI serving endpoint",
|
||||||
)
|
)
|
||||||
# api_token: Optional[SecretStr] = Field(
|
|
||||||
# default=None,
|
|
||||||
# description="A bearer token if your TGI endpoint is protected.",
|
|
||||||
# )
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def sample_run_config(cls, url: str = "${env.TGI_URL}", **kwargs):
|
def sample_run_config(cls, url: str = "${env.TGI_URL}", **kwargs):
|
||||||
|
|
|
||||||
|
|
@ -225,7 +225,6 @@ def test_text_chat_completion_with_tool_calling_and_non_streaming(
|
||||||
tool_prompt_format=provider_tool_format,
|
tool_prompt_format=provider_tool_format,
|
||||||
stream=False,
|
stream=False,
|
||||||
)
|
)
|
||||||
print(response)
|
|
||||||
# No content is returned for the system message since we expect the
|
# No content is returned for the system message since we expect the
|
||||||
# response to be a tool call
|
# response to be a tool call
|
||||||
assert response.completion_message.content == ""
|
assert response.completion_message.content == ""
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue