fix(forge, agent): Resolve several issues for Chinese language & region

autogpt/autogpt/app/agent_protocol_server.py

@@ -6,7 +6,8 @@
 from uuid import uuid4
 
 import orjson
-from fastapi import APIRouter, FastAPI, UploadFile
+from typing import Callable
+from fastapi import APIRouter, FastAPI, UploadFile, FastAPI, Request, Response
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import RedirectResponse, StreamingResponse
 from fastapi.staticfiles import StaticFiles
@@ -40,6 +41,15 @@
 logger = logging.getLogger(__name__)
 
 
+def create_middleware(app: FastAPI):
+    async def add_custom_header(request: Request, call_next: Callable) -> Response:
+        response = await call_next(request)
+        response.headers["Content-Type"] = "application/json ; charset=UTF-8"
+        return response
+
+    app.middleware("http")(add_custom_header)
+
+
 class AgentProtocolServer:
     _task_budgets: dict[str, ModelProviderBudget]
 
@@ -76,7 +86,7 @@
             "Modified version of The Agent Protocol.",
             version="v0.4",
         )
-
+        create_middleware(app)
         # Configure CORS middleware
         default_origins = [f"http://localhost:{port}"]  # Default only local access
         configured_origins = [
@@ -88,7 +98,7 @@
 
         app.add_middleware(
             CORSMiddleware,
-            allow_origins=origins,
+            allow_origins=["*"],
             allow_credentials=True,
             allow_methods=["*"],
             allow_headers=["*"],

forge/forge/agent_protocol/models/task.py

@@ -17,7 +17,7 @@ class TaskRequestBody(BaseModel):
         description="Input prompt for the task.",
         example="Write the words you receive to the file 'output.txt'.",
     )
-    additional_input: dict[str, Any] = Field(default_factory=dict)
+    additional_input: Optional[dict[str, Any]] = Field(default_factory=dict)
 
 
 class Task(TaskRequestBody):
@@ -55,7 +55,7 @@ class StepRequestBody(BaseModel):
     input: str = Field(
         ..., description="Input prompt for the step.", example="Washington"
     )
-    additional_input: dict[str, Any] = Field(default_factory=dict)
+    additional_input: Optional[dict[str, Any]] = Field(default_factory=dict)
 
 
 class StepStatus(Enum):

forge/forge/config/config.py

@@ -210,7 +210,7 @@
 
 def assert_config_has_openai_api_key(config: Config) -> None:
     """Check if the OpenAI API key is set in config.py or as an environment variable."""
-    key_pattern = r"^sk-(proj-)?\w{48}"
+    key_pattern = r"^sk-(?:proj-|wa-)?\w{48}"
     openai_api_key = (
         config.openai_credentials.api_key.get_secret_value()
         if config.openai_credentials