fix(agent-zero): auto-date in system prompt, env vars for config
This commit is contained in:
@@ -1,15 +1,16 @@
|
|||||||
"""A2A agent for kagentz — uvicorn-based, calls LiteLLM for responses."""
|
"""A2A server for kagentz — calls LiteLLM directly for responses."""
|
||||||
import os, json, uuid, threading, asyncio
|
import os, json, uuid, asyncio
|
||||||
|
from datetime import datetime
|
||||||
import uvicorn
|
import uvicorn
|
||||||
from fastapi import FastAPI, Request
|
from fastapi import FastAPI, Request
|
||||||
from fastapi.responses import JSONResponse
|
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
app = FastAPI()
|
app = FastAPI()
|
||||||
task_results = {}
|
task_results = {}
|
||||||
LITELLM_URL = "https://litellm.sysloggh.net/v1"
|
LITELLM_URL = os.getenv("LITELLM_URL", "https://litellm.sysloggh.net/v1")
|
||||||
LITELLM_KEY = "sk-U_ydi3B-wfGU-_xESkoU1Q"
|
LITELLM_KEY = os.getenv("LITELLM_KEY", "sk-akDPhdO8qlhYNtVet6KZog")
|
||||||
LITELLM_MODEL = "syslog-auto"
|
LITELLM_MODEL = os.getenv("LITELLM_MODEL", "syslog-auto")
|
||||||
|
NOW = datetime.utcnow().strftime("%Y-%m-%d %H:%M UTC")
|
||||||
|
|
||||||
@app.get("/.well-known/agent.json")
|
@app.get("/.well-known/agent.json")
|
||||||
async def agent_card():
|
async def agent_card():
|
||||||
@@ -59,7 +60,7 @@ async def call_litellm(task_id, message):
|
|||||||
json={
|
json={
|
||||||
"model": LITELLM_MODEL,
|
"model": LITELLM_MODEL,
|
||||||
"messages": [
|
"messages": [
|
||||||
{"role": "system", "content": "You are kagentz, an Agent Zero instance on CT 105. You communicate via Zulip. Keep responses concise."},
|
{"role": "system", "content": f"You are kagentz, an Agent Zero instance. Current date: {NOW}. Keep responses concise and accurate."},
|
||||||
{"role": "user", "content": message},
|
{"role": "user", "content": message},
|
||||||
],
|
],
|
||||||
"max_tokens": 2000,
|
"max_tokens": 2000,
|
||||||
@@ -74,4 +75,5 @@ async def call_litellm(task_id, message):
|
|||||||
task_results[task_id] = ("failed", str(e))
|
task_results[task_id] = ("failed", str(e))
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
print(f"kagentz A2A server on port 8001 (model={LITELLM_MODEL}, date={NOW})")
|
||||||
uvicorn.run(app, host="0.0.0.0", port=8001, log_level="info")
|
uvicorn.run(app, host="0.0.0.0", port=8001, log_level="info")
|
||||||
|
|||||||
Reference in New Issue
Block a user