Niansuh commited on
Commit
aac62ed
·
verified ·
1 Parent(s): 3b4d658

Update api/utils.py

Browse files
Files changed (1) hide show
  1. api/utils.py +8 -2
api/utils.py CHANGED
@@ -4,7 +4,13 @@ from typing import Any, Dict, Optional
4
  import uuid
5
 
6
  import httpx
7
- from api.config import MODEL_MAPPING, AGENT_MODE, TRENDING_AGENT_MODE, headers
 
 
 
 
 
 
8
  from fastapi import HTTPException
9
  from api.models import ChatRequest
10
  from api.logger import setup_logger
@@ -79,7 +85,7 @@ async def process_streaming_response(request: ChatRequest):
79
  try:
80
  async with client.stream(
81
  "POST",
82
- f"{BASE_URL}/api/chat",
83
  headers=headers,
84
  json=json_data,
85
  timeout=100,
 
4
  import uuid
5
 
6
  import httpx
7
+ from api.config import (
8
+ MODEL_MAPPING,
9
+ AGENT_MODE,
10
+ TRENDING_AGENT_MODE,
11
+ headers,
12
+ BASE_URL, # Import BASE_URL
13
+ )
14
  from fastapi import HTTPException
15
  from api.models import ChatRequest
16
  from api.logger import setup_logger
 
85
  try:
86
  async with client.stream(
87
  "POST",
88
+ f"{BASE_URL}/api/chat", # BASE_URL is now defined
89
  headers=headers,
90
  json=json_data,
91
  timeout=100,