Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import os
|
2 |
import sys
|
3 |
import json
|
@@ -9,7 +10,7 @@ from typing import List, Dict, Optional, Tuple
|
|
9 |
from enum import Enum
|
10 |
|
11 |
from fastapi import FastAPI, HTTPException
|
12 |
-
from
|
13 |
from fastapi.middleware.cors import CORSMiddleware
|
14 |
from pydantic import BaseModel
|
15 |
import asyncio
|
@@ -23,7 +24,7 @@ logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(
|
|
23 |
logger = logging.getLogger("TxAgentAPI")
|
24 |
|
25 |
# App
|
26 |
-
app = FastAPI(title="TxAgent API", version="2.2.1") # Version
|
27 |
|
28 |
app.add_middleware(
|
29 |
CORSMiddleware,
|
@@ -340,4 +341,4 @@ async def chat_stream_endpoint(request: ChatRequest):
|
|
340 |
logger.error(f"Streaming error: {e}")
|
341 |
yield f"⚠️ Error: {e}"
|
342 |
|
343 |
-
return StreamingResponse(token_stream(), media_type="text/plain")
|
|
|
1 |
+
```python
|
2 |
import os
|
3 |
import sys
|
4 |
import json
|
|
|
10 |
from enum import Enum
|
11 |
|
12 |
from fastapi import FastAPI, HTTPException
|
13 |
+
from fastapi.responses import StreamingResponse
|
14 |
from fastapi.middleware.cors import CORSMiddleware
|
15 |
from pydantic import BaseModel
|
16 |
import asyncio
|
|
|
24 |
logger = logging.getLogger("TxAgentAPI")
|
25 |
|
26 |
# App
|
27 |
+
app = FastAPI(title="TxAgent API", version="2.2.1") # Version for hash-based analysis
|
28 |
|
29 |
app.add_middleware(
|
30 |
CORSMiddleware,
|
|
|
341 |
logger.error(f"Streaming error: {e}")
|
342 |
yield f"⚠️ Error: {e}"
|
343 |
|
344 |
+
return StreamingResponse(token_stream(), media_type="text/plain")
|