forked from phoenix-oss/llama-stack-mirror
fix security update
This commit is contained in:
parent
8943b283e9
commit
d851df055a
14 changed files with 48 additions and 41 deletions
|
@ -12,6 +12,8 @@ from llama_stack.distribution.ui.app import get_access_token
|
|||
|
||||
class LlamaStackApi:
|
||||
def __init__(self):
|
||||
print("NEW CLIENT")
|
||||
print(get_access_token())
|
||||
self.client = LlamaStackClient(
|
||||
api_key=get_access_token(),
|
||||
base_url=os.environ.get("LLAMA_STACK_ENDPOINT", "http://localhost:8321"),
|
||||
|
@ -30,5 +32,3 @@ class LlamaStackApi:
|
|||
scoring_params = {fn_id: None for fn_id in scoring_function_ids}
|
||||
return self.client.scoring.score(input_rows=[row], scoring_functions=scoring_params)
|
||||
|
||||
|
||||
llama_stack_api = LlamaStackApi()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue