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."""
|
||||
import os, json, uuid, threading, asyncio
|
||||
"""A2A server for kagentz — calls LiteLLM directly for responses."""
|
||||
import os, json, uuid, asyncio
|
||||
from datetime import datetime
|
||||
import uvicorn
|
||||
from fastapi import FastAPI, Request
|
||||
from fastapi.responses import JSONResponse
|
||||
import httpx
|
||||
|
||||
app = FastAPI()
|
||||
task_results = {}
|
||||
LITELLM_URL = "https://litellm.sysloggh.net/v1"
|
||||
LITELLM_KEY = "sk-U_ydi3B-wfGU-_xESkoU1Q"
|
||||
LITELLM_MODEL = "syslog-auto"
|
||||
LITELLM_URL = os.getenv("LITELLM_URL", "https://litellm.sysloggh.net/v1")
|
||||
LITELLM_KEY = os.getenv("LITELLM_KEY", "sk-akDPhdO8qlhYNtVet6KZog")
|
||||
LITELLM_MODEL = os.getenv("LITELLM_MODEL", "syslog-auto")
|
||||
NOW = datetime.utcnow().strftime("%Y-%m-%d %H:%M UTC")
|
||||
|
||||
@app.get("/.well-known/agent.json")
|
||||
async def agent_card():
|
||||
@@ -59,7 +60,7 @@ async def call_litellm(task_id, message):
|
||||
json={
|
||||
"model": LITELLM_MODEL,
|
||||
"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},
|
||||
],
|
||||
"max_tokens": 2000,
|
||||
@@ -74,4 +75,5 @@ async def call_litellm(task_id, message):
|
||||
task_results[task_id] = ("failed", str(e))
|
||||
|
||||
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")
|
||||
|
||||
Reference in New Issue
Block a user