mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 12:07:34 +00:00
nuke safety/list_shields, we don't need it now
This commit is contained in:
parent
bc394882a0
commit
484dc2e4f5
3 changed files with 2 additions and 51 deletions
|
@ -13,10 +13,11 @@ import fire
|
|||
import httpx
|
||||
|
||||
from llama_models.llama3.api.datatypes import * # noqa: F403
|
||||
from llama_stack.distribution.datatypes import RemoteProviderConfig
|
||||
from pydantic import BaseModel
|
||||
from termcolor import cprint
|
||||
|
||||
from llama_stack.distribution.datatypes import RemoteProviderConfig
|
||||
|
||||
from llama_stack.apis.safety import * # noqa: F403
|
||||
|
||||
|
||||
|
@ -61,24 +62,6 @@ class SafetyClient(Safety):
|
|||
content = response.json()
|
||||
return RunShieldResponse(**content)
|
||||
|
||||
async def list_shields(self) -> ListShieldsResponse:
|
||||
async with httpx.AsyncClient() as client:
|
||||
response = await client.post(
|
||||
f"{self.base_url}/safety/list_shields",
|
||||
json={},
|
||||
headers={"Content-Type": "application/json"},
|
||||
timeout=20,
|
||||
)
|
||||
|
||||
if response.status_code != 200:
|
||||
content = await response.aread()
|
||||
error = f"Error: HTTP {response.status_code} {content.decode()}"
|
||||
cprint(error, "red")
|
||||
raise Exception(error)
|
||||
|
||||
content = response.json()
|
||||
return ListShieldsResponse(**content)
|
||||
|
||||
|
||||
async def run_main(host: str, port: int):
|
||||
client = SafetyClient(f"http://{host}:{port}")
|
||||
|
@ -100,9 +83,6 @@ async def run_main(host: str, port: int):
|
|||
)
|
||||
print(response)
|
||||
|
||||
response = await client.list_shields()
|
||||
print(response)
|
||||
|
||||
|
||||
def main(host: str, port: int):
|
||||
asyncio.run(run_main(host, port))
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue