use archfc hosted on aws (#409)

katanemo · Feb 13, 2025 · 4ec03af · 4ec03af
1 parent 28ee14a
commit 4ec03af
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 9 deletions.
diff --git a/model_server/src/commons/globals.py b/model_server/src/commons/globals.py
@@ -15,7 +15,7 @@
 
 
 # Define the client
-ARCH_ENDPOINT = os.getenv("ARCH_ENDPOINT", "https://api.fc.archgw.com/v1")
+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)
 

diff --git a/model_server/src/main.py b/model_server/src/main.py
@@ -4,7 +4,7 @@
 import logging
 import src.commons.utils as utils
 
-from src.commons.globals import handler_map
+from src.commons.globals import ARCH_ENDPOINT, handler_map
 from src.core.utils.model_utils import (
     ChatMessage,
     ChatCompletionResponse,
@@ -51,6 +51,8 @@
 app = FastAPI()
 FastAPIInstrumentor().instrument_app(app)
 
+logger.info(f"using archfc endpoint: {ARCH_ENDPOINT}")
+
 
 @app.get("/healthz")
 async def healthz():

diff --git a/tests/rest/api_model_server.rest b/tests/rest/api_model_server.rest
@@ -1,6 +1,5 @@
 @model_server_endpoint = http://localhost:51000
-@archfc_endpoint = https://api.fc.archgw.com
-
+@archfc_endpoint = https://archfc.katanemo.dev
 
 ### talk to function calling endpoint
 POST {{model_server_endpoint}}/function_calling HTTP/1.1
@@ -119,7 +118,7 @@ Content-Type: application/json
 }
 
 ### talk to Arch-Intent directly for completion
-POST {{archfc_endpoint}}/v1/chat/completions HTTP/1.1
+POST {{{{archfc_endpoint}}}}/v1/chat/completions HTTP/1.1
 Content-Type: application/json
 
 {

diff --git a/tests/rest/insurance_agent.rest b/tests/rest/insurance_agent.rest
@@ -1,5 +1,5 @@
 @model_server_endpoint = http://localhost:51000
-@archfc_endpoint = https://api.fc.archgw.com
+@archfc_endpoint = https://archfc.katanemo.dev
 
 ### multi turn conversation with intent, except parameter gathering
 
@@ -55,7 +55,7 @@ Content-Type: application/json
   ]
 }
 ### talk to Arch-Intent directly for completion
-POST https://api.fc.archgw.com/v1/chat/completions HTTP/1.1
+POST https://archfc.katanemo.dev/v1/chat/completions HTTP/1.1
 Content-Type: application/json
 
 {
@@ -126,7 +126,7 @@ Content-Type: application/json
   ]
 }
 ### talk to Arch-Intent directly for completion, expect No
-POST https://api.fc.archgw.com/v1/chat/completions HTTP/1.1
+POST https://archfc.katanemo.dev/v1/chat/completions HTTP/1.1
 Content-Type: application/json
 
 {

diff --git a/tests/rest/network_agent.rest b/tests/rest/network_agent.rest
@@ -1,5 +1,5 @@
 @model_server_endpoint = http://localhost:51000
-@archfc_endpoint = https://api.fc.archgw.com
+@archfc_endpoint = https://archfc.katanemo.dev
 
 ### single turn function calling all parameters insurance agent summary