apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-23 15:38:07 +02:00
Code Issues Projects Releases Packages Wiki Activity

refactor

Browse source
This commit is contained in:
Adil Hafeez 2025-03-14 17:34:18 -07:00
parent f4462d1ed5
commit 816f513663
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
5 changed files with 2 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

2
model_server/src/commons/globals.py
View file

@ -21,13 +21,11 @@ ARCH_ENDPOINT = os.getenv("ARCH_ENDPOINT", "https://archfc.katanemo.dev/v1")
ARCH_API_KEY = "EMPTY"
ARCH_CLIENT = OpenAI(base_url=ARCH_ENDPOINT, api_key=ARCH_API_KEY)
ARCH_AGENT_CLIENT = ARCH_CLIENT
# ARCH_AGENT_CLIENT = OpenAI(api_key=os.getenv("OPENAI_API_KEY", "EMPTY"))
# Define model names
ARCH_INTENT_MODEL_ALIAS = "Arch-Intent"
ARCH_FUNCTION_MODEL_ALIAS = "Arch-Function"
ARCH_AGENT_MODEL_ALIAS = ARCH_FUNCTION_MODEL_ALIAS
# ARCH_AGENT_MODEL_ALIAS = "gpt-4o-mini"
ARCH_GUARD_MODEL_ALIAS = "katanemo/Arch-Guard"
# Define model handlers

1
model_server/src/core/function_calling.py
View file

@ -630,7 +630,6 @@ class ArchFunctionHandler(ArchBaseHandler):
return chat_completion_response
# override ArchFunctionHandler
class ArchAgentHandler(ArchFunctionHandler):
def __init__(self, client: OpenAI, model_name: str, config: ArchAgentConfig):
super().__init__(client, model_name, config)