using System; using System.Collections.Generic; using System.IO; using System.Net.Http; using System.Net.Http.Headers; using System.Text; using System.Text.Json; using System.Threading; using System.Threading.Tasks; using Gott.Models; using Microsoft.AspNetCore.Http; namespace Gott.Services; public class AnthropicService { private readonly IHttpClientFactory _httpFactory; private readonly string _token; private readonly ToolService _toolService; private readonly DatabaseService _db; private readonly MemoryService _memoryService; public AnthropicService(IHttpClientFactory httpFactory, string token, ToolService toolService, DatabaseService db, MemoryService memoryService = null) { _httpFactory = httpFactory; _token = token; _toolService = toolService; _db = db; _memoryService = memoryService; } public async Task StreamChat( HttpResponse response, string modelId, List systemPrompt, List messages, string chainId, string workspace, string sessionId = null, string agentId = null, bool useMemory = true, string webSearchVersion = null, int maxIterations = 100, string toolChoice = "any", CancellationToken cancellationToken = default) { var sw = System.Diagnostics.Stopwatch.StartNew(); var usage = new TokenUsage(); int iterations = 0; string lastStopReason = ""; var chain = new ChainRecord { ChainId = chainId, SessionId = sessionId, Model = modelId }; try { while (iterations < maxIterations) { iterations++; var payload = new Dictionary { ["model"] = modelId, ["max_tokens"] = 16384, ["cache_control"] = new { type = "ephemeral" }, ["messages"] = messages, ["stream"] = true }; if (systemPrompt != null && systemPrompt.Count > 0) payload["system"] = systemPrompt; var hasWorkspace = !string.IsNullOrEmpty(workspace); var includeMemory = useMemory && _memoryService != null && !string.IsNullOrEmpty(agentId); var hasTools = hasWorkspace || includeMemory; var hasWebSearch = !string.IsNullOrEmpty(webSearchVersion); if (hasTools || hasWebSearch) { var tools = new List(); if (hasTools) tools.AddRange(_toolService.GetToolDefinitions(includeFileTools: hasWorkspace, includeMemory: includeMemory)); if (hasWebSearch) tools.Add(new { type = webSearchVersion, name = "web_search", max_uses = 5 }); payload["tools"] = tools; // "any" erzwingt Tool-Nutzung, aber nur in der ersten Iteration, // damit Claude nicht endlos Tools aufruft wenn die Aufgabe erledigt ist. var effectiveChoice = (toolChoice == "any" && iterations == 1) ? "any" : "auto"; payload["tool_choice"] = new { type = effectiveChoice }; } var client = CreateClient(); var json = JsonSerializer.Serialize(payload, new JsonSerializerOptions { WriteIndented = true }); LogPayload(json, sessionId, chainId, iterations); var httpContent = new StringContent(json, Encoding.UTF8, "application/json"); var httpResp = await client.SendAsync( new HttpRequestMessage(HttpMethod.Post, "https://api.anthropic.com/v1/messages") { Content = httpContent }, HttpCompletionOption.ResponseHeadersRead, cancellationToken ); if (!httpResp.IsSuccessStatusCode) { var errBody = await httpResp.Content.ReadAsStringAsync(cancellationToken); var errMsg = $"API {(int)httpResp.StatusCode}: {errBody}"; await _db.SaveMessage(new MessageRecord { ChainId = chainId, Iteration = iterations, Role = "system", Type = "error", Content = errMsg, IsError = true }); await TryWriteSse(response, "error", new { error = errMsg }, cancellationToken); await FinalizeChain(chain, usage, iterations, "error", sw); await TryWriteSse(response, "done", new { chainId = chain.ChainId }, cancellationToken); return; } var (stopReason, assistantContent, executedTools, streamErrorInfo) = await ProcessStream(response, httpResp, usage, workspace, agentId, cancellationToken); lastStopReason = stopReason; messages.Add(new { role = "assistant", content = assistantContent }); // Assistant-Blöcke persistieren foreach (var block in assistantContent) { var blockJson = JsonSerializer.Serialize(block); var el = JsonSerializer.Deserialize(blockJson); var blockType = el.GetProperty("type").GetString() ?? ""; if (blockType == "text") { await _db.SaveMessage(new MessageRecord { ChainId = chainId, Iteration = iterations, Role = "assistant", Type = "text", Content = el.GetProperty("text").GetString() }); } else if (blockType == "tool_use") { await _db.SaveMessage(new MessageRecord { ChainId = chainId, Iteration = iterations, Role = "assistant", Type = "tool_use", ToolName = el.GetProperty("name").GetString(), ToolCallId = el.GetProperty("id").GetString(), Content = el.GetProperty("input").ToString() }); } } // Tool Results persistieren foreach (var tr in executedTools) { await _db.SaveMessage(new MessageRecord { ChainId = chainId, Iteration = iterations, Role = "user", Type = "tool_result", ToolCallId = tr.Id, Content = tr.Result, IsError = tr.IsError }); } if (stopReason != "tool_use") { var finalStopReason = string.IsNullOrEmpty(stopReason) ? "unexpected_end" : stopReason; await FinalizeChain(chain, usage, iterations, finalStopReason, sw, streamErrorInfo); await TryWriteSse(response, "done", new { model = modelId, inputTokens = usage.InputTokens, outputTokens = usage.OutputTokens, cacheCreationTokens = usage.CacheCreationTokens, cacheReadTokens = usage.CacheReadTokens, iterations, chainId, durationMs = sw.ElapsedMilliseconds, stopReason = finalStopReason, streamErrorInfo }, cancellationToken); return; } var toolResults = new List(); foreach (var tr in executedTools) { // Tool-Results in der History auf 20.000 Zeichen kürzen. // Das vollständige Ergebnis wurde bereits in der DB gespeichert. // 20k Zeichen ≈ ~5k Tokens – verhindert Token-Explosion bei großen // read_file / execute-Outputs die sich über alle weiteren Iterationen schleppen. const int MaxToolResultChars = 20_000; var content = tr.Result.Length > MaxToolResultChars ? tr.Result[..MaxToolResultChars] + $"\n\n[... gekürzt, {tr.Result.Length - MaxToolResultChars} weitere Zeichen ...]" : tr.Result; toolResults.Add(new { type = "tool_result", tool_use_id = tr.Id, content, is_error = tr.IsError }); } messages.Add(new { role = "user", content = toolResults }); } await FinalizeChain(chain, usage, iterations, "max_iterations", sw); await TryWriteSse(response, "error", new { error = "Max iterations erreicht" }, cancellationToken); await TryWriteSse(response, "done", new { chainId = chain.ChainId }, cancellationToken); } catch (OperationCanceledException) { // Client hat die Verbindung getrennt (z.B. Browser-Refresh) await FinalizeChain(chain, usage, iterations, "client_disconnected", sw); } catch (Exception ex) { // Unerwarteter Fehler – in DB speichern und versuchen an Client zu senden var errMsg = $"Stream-Fehler: {ex.Message}"; await _db.SaveMessage(new MessageRecord { ChainId = chainId, Iteration = iterations, Role = "system", Type = "error", Content = errMsg, IsError = true }); await FinalizeChain(chain, usage, iterations, "error", sw); await TryWriteSse(response, "error", new { error = errMsg }, cancellationToken); await TryWriteSse(response, "done", new { chainId = chain.ChainId }, cancellationToken); } } /// /// Non-Streaming Chat für Telegram. Gibt den finalen Antworttext zurück. /// onIteration wird nach jeder Iteration aufgerufen (z.B. für Typing-Indicator). /// public async Task ChatNonStreaming( ChatContext ctx, List messages, Func onIteration = null) { var sw = System.Diagnostics.Stopwatch.StartNew(); var usage = new TokenUsage(); int iterations = 0; var finalText = new StringBuilder(); var chain = new ChainRecord { ChainId = ctx.ChainId, SessionId = ctx.SessionId, Model = ctx.ModelId }; try { while (iterations < ctx.MaxIterations) { iterations++; onIteration?.Invoke(iterations); var payload = new Dictionary { ["model"] = ctx.ModelId, ["max_tokens"] = 16384, ["messages"] = messages, ["stream"] = false }; if (ctx.SystemPrompt != null && ctx.SystemPrompt.Count > 0) payload["system"] = ctx.SystemPrompt; var hasWorkspace = !string.IsNullOrEmpty(ctx.Workspace); var includeMemory = ctx.UseMemory && _memoryService != null && !string.IsNullOrEmpty(ctx.AgentId); var hasTools = hasWorkspace || includeMemory; var hasWebSearch = !string.IsNullOrEmpty(ctx.WebSearchVersion); if (hasTools || hasWebSearch) { var tools = new List(); if (hasTools) tools.AddRange(_toolService.GetToolDefinitions(includeFileTools: hasWorkspace, includeMemory: includeMemory)); if (hasWebSearch) tools.Add(new { type = ctx.WebSearchVersion, name = "web_search", max_uses = 5 }); payload["tools"] = tools; var effectiveChoice = (ctx.ToolChoice == "any" && iterations == 1) ? "any" : "auto"; payload["tool_choice"] = new { type = effectiveChoice }; } var client = CreateClient(); var json = JsonSerializer.Serialize(payload, new JsonSerializerOptions { WriteIndented = true }); LogPayload(json, ctx.SessionId, ctx.ChainId, iterations); var httpContent = new StringContent(json, Encoding.UTF8, "application/json"); var httpResp = await client.SendAsync( new HttpRequestMessage(HttpMethod.Post, "https://api.anthropic.com/v1/messages") { Content = httpContent }); var respBody = await httpResp.Content.ReadAsStringAsync(); if (!httpResp.IsSuccessStatusCode) { var errMsg = $"API {(int)httpResp.StatusCode}: {respBody}"; await _db.SaveMessage(new MessageRecord { ChainId = ctx.ChainId, Iteration = iterations, Role = "system", Type = "error", Content = errMsg, IsError = true }); await FinalizeChain(chain, usage, iterations, "error", sw); throw new Exception($"Claude API Fehler: {(int)httpResp.StatusCode}"); } var resp = JsonSerializer.Deserialize(respBody); // Usage tracken if (resp.TryGetProperty("usage", out var u)) { if (u.TryGetProperty("input_tokens", out var it)) usage.InputTokens += it.GetInt64(); if (u.TryGetProperty("output_tokens", out var ot)) usage.OutputTokens += ot.GetInt64(); if (u.TryGetProperty("cache_creation_input_tokens", out var cc)) usage.CacheCreationTokens += cc.GetInt64(); if (u.TryGetProperty("cache_read_input_tokens", out var cr)) usage.CacheReadTokens += cr.GetInt64(); } var stopReason = resp.TryGetProperty("stop_reason", out var sr) ? sr.GetString() ?? "" : ""; var content = resp.GetProperty("content"); var assistantContent = new List(); var toolResults = new List(); foreach (var block in content.EnumerateArray()) { var blockType = block.GetProperty("type").GetString() ?? ""; if (blockType == "text") { var text = block.GetProperty("text").GetString() ?? ""; assistantContent.Add(new { type = "text", text }); finalText.Append(text); await _db.SaveMessage(new MessageRecord { ChainId = ctx.ChainId, Iteration = iterations, Role = "assistant", Type = "text", Content = text }); } else if (blockType == "tool_use") { var toolId = block.GetProperty("id").GetString() ?? ""; var toolName = block.GetProperty("name").GetString() ?? ""; var toolInput = block.GetProperty("input"); assistantContent.Add(new { type = "tool_use", id = toolId, name = toolName, input = toolInput }); await _db.SaveMessage(new MessageRecord { ChainId = ctx.ChainId, Iteration = iterations, Role = "assistant", Type = "tool_use", Content = toolInput.GetRawText(), ToolName = toolName, ToolCallId = toolId }); string toolResult; bool isError = false; try { toolResult = await _toolService.ExecuteTool(toolName, toolInput, ctx.Workspace, _memoryService, ctx.AgentId); } catch (Exception ex) { toolResult = $"Fehler: {ex.Message}"; isError = true; } // Tool-Result in der History kürzen (wie bei StreamChat) const int MaxToolResultChars = 20_000; var truncatedResult = toolResult.Length > MaxToolResultChars ? toolResult[..MaxToolResultChars] + $"\n\n[... gekürzt, {toolResult.Length - MaxToolResultChars} weitere Zeichen ...]" : toolResult; toolResults.Add(new { type = "tool_result", tool_use_id = toolId, content = truncatedResult, is_error = isError }); await _db.SaveMessage(new MessageRecord { ChainId = ctx.ChainId, Iteration = iterations, Role = "user", Type = "tool_result", Content = toolResult, ToolName = toolName, ToolCallId = toolId, IsError = isError }); } } messages.Add(new Dictionary { ["role"] = "assistant", ["content"] = assistantContent }); if (stopReason == "end_turn" || stopReason == "stop_sequence" || toolResults.Count == 0) { await FinalizeChain(chain, usage, iterations, stopReason, sw); break; } messages.Add(new Dictionary { ["role"] = "user", ["content"] = toolResults }); } } catch { if (iterations > 0) await FinalizeChain(chain, usage, iterations, "error", sw); throw; } return finalText.Length > 0 ? finalText.ToString() : "(Keine Textantwort)"; } private static void LogPayload(string json, string sessionId, string chainId, int iteration) { try { var logsDir = Path.Combine(AppContext.BaseDirectory, "data", "logs"); Directory.CreateDirectory(logsDir); var timestamp = DateTimeOffset.UtcNow.ToUnixTimeSeconds(); var fileName = $"{timestamp}-{sessionId}-{chainId}-iter{iteration}-payload.json"; File.WriteAllText(Path.Combine(logsDir, fileName), json); } catch { /* Logging darf nie die App crashen */ } } private async Task FinalizeChain(ChainRecord chain, TokenUsage usage, int iterations, string stopReason, System.Diagnostics.Stopwatch sw, string streamErrorInfo = null) { chain.InputTokens = usage.InputTokens; chain.OutputTokens = usage.OutputTokens; chain.CacheCreationTokens = usage.CacheCreationTokens; chain.CacheReadTokens = usage.CacheReadTokens; chain.Iterations = iterations; chain.StopReason = stopReason; chain.StreamErrorInfo = streamErrorInfo; chain.DurationMs = sw.ElapsedMilliseconds; await _db.UpdateChain(chain); } private async Task<(string StopReason, List Content, List Tools, string StreamErrorInfo)> ProcessStream( HttpResponse response, HttpResponseMessage httpResp, TokenUsage usage, string workspace, string agentId = null, CancellationToken cancellationToken = default) { var stream = await httpResp.Content.ReadAsStreamAsync(cancellationToken); using var reader = new StreamReader(stream); var stopReason = ""; var assistantContent = new List(); var currentToolId = ""; var currentToolName = ""; var currentToolInputJson = new StringBuilder(); var currentTextBlock = new StringBuilder(); var blockTypes = new Dictionary(); var webSearchResultBlocks = new Dictionary(); var executedTools = new List(); var streamErrors = new List(); string line; while ((line = await reader.ReadLineAsync(cancellationToken)) != null) { if (!line.StartsWith("data: ")) continue; var data = line["data: ".Length..]; if (string.IsNullOrWhiteSpace(data)) continue; JsonElement evt; try { evt = JsonSerializer.Deserialize(data); } catch { continue; } var evtType = evt.TryGetProperty("type", out var t) ? t.GetString() ?? "" : ""; switch (evtType) { case "message_start": { if (evt.TryGetProperty("message", out var msg) && msg.TryGetProperty("usage", out var u)) { if (u.TryGetProperty("input_tokens", out var it)) usage.InputTokens += it.GetInt64(); if (u.TryGetProperty("cache_creation_input_tokens", out var cc)) usage.CacheCreationTokens += cc.GetInt64(); if (u.TryGetProperty("cache_read_input_tokens", out var cr)) usage.CacheReadTokens += cr.GetInt64(); } break; } case "content_block_start": { var idx = evt.GetProperty("index").GetInt32(); var cb = evt.GetProperty("content_block"); var cbType = cb.GetProperty("type").GetString() ?? ""; blockTypes[idx] = cbType; if (cbType == "text") { currentTextBlock.Clear(); await WriteSseEvent(response, "text_start", new { index = idx }); } else if (cbType == "tool_use") { if (currentTextBlock.Length > 0) { assistantContent.Add(new { type = "text", text = currentTextBlock.ToString() }); currentTextBlock.Clear(); } currentToolId = cb.GetProperty("id").GetString() ?? ""; currentToolName = cb.GetProperty("name").GetString() ?? ""; currentToolInputJson.Clear(); await WriteSseEvent(response, "tool_start", new { name = currentToolName, id = currentToolId }); } else if (cbType == "server_tool_use") { if (currentTextBlock.Length > 0) { assistantContent.Add(new { type = "text", text = currentTextBlock.ToString() }); currentTextBlock.Clear(); } currentToolId = cb.GetProperty("id").GetString() ?? ""; currentToolName = cb.GetProperty("name").GetString() ?? ""; currentToolInputJson.Clear(); await WriteSseEvent(response, "web_search_start", new { name = currentToolName, id = currentToolId }); } else if (cbType == "web_search_tool_result") { // Gesamten Block als JSON speichern für die Conversation-History var rawBlock = JsonSerializer.Deserialize(cb.GetRawText()); webSearchResultBlocks[idx] = rawBlock; } break; } case "content_block_delta": { var delta = evt.GetProperty("delta"); var deltaType = delta.GetProperty("type").GetString() ?? ""; if (deltaType == "text_delta") { var text = delta.GetProperty("text").GetString() ?? ""; currentTextBlock.Append(text); await WriteSseEvent(response, "text_delta", new { text }); } else if (deltaType == "input_json_delta") { var partial = delta.GetProperty("partial_json").GetString() ?? ""; currentToolInputJson.Append(partial); } break; } case "content_block_stop": { var idx = evt.GetProperty("index").GetInt32(); if (blockTypes.TryGetValue(idx, out var bt)) { if (bt == "text") { var text = currentTextBlock.ToString(); if (text.Length > 0) { assistantContent.Add(new { type = "text", text }); await WriteSseEvent(response, "text_end", new { }); } currentTextBlock.Clear(); } else if (bt == "tool_use") { JsonElement inputEl; try { inputEl = JsonSerializer.Deserialize(currentToolInputJson.ToString()); } catch { inputEl = JsonSerializer.Deserialize("{}"); } assistantContent.Add(new { type = "tool_use", id = currentToolId, name = currentToolName, input = inputEl }); string toolResult; bool isError = false; try { toolResult = await _toolService.ExecuteTool(currentToolName, inputEl, workspace, _memoryService, agentId); } catch (Exception ex) { toolResult = $"Fehler: {ex.Message}"; isError = true; } executedTools.Add(new ToolResult(currentToolId, toolResult, isError)); await WriteSseEvent(response, "tool_end", new { name = currentToolName, input = currentToolInputJson.ToString(), result = toolResult.Length > 500 ? toolResult[..500] + "..." : toolResult, isError }); } else if (bt == "server_tool_use") { // Server-side tool (Web Search) – Anthropic führt es aus, wir speichern nur die Anfrage JsonElement inputEl; try { inputEl = JsonSerializer.Deserialize(currentToolInputJson.ToString()); } catch { inputEl = JsonSerializer.Deserialize("{}"); } assistantContent.Add(new { type = "server_tool_use", id = currentToolId, name = currentToolName, input = inputEl }); } else if (bt == "web_search_tool_result") { // Web-Search-Ergebnis von Anthropic – als Raw-JSON in assistantContent aufnehmen assistantContent.Add(webSearchResultBlocks[idx]); await WriteSseEvent(response, "web_search_end", new { id = currentToolId }); } } break; } case "message_delta": { if (evt.TryGetProperty("delta", out var d)) { if (d.TryGetProperty("stop_reason", out var sr)) { stopReason = sr.GetString() ?? ""; // Bei problematischen stop_reasons Debug-Infos sammeln if (stopReason is "unexpected_end" or "max_tokens") { var debugInfo = new Dictionary { ["event"] = "message_delta", ["stop_reason"] = stopReason, ["output_tokens"] = usage.OutputTokens, ["text_length"] = currentTextBlock.Length, ["timestamp"] = DateTime.UtcNow.ToString("o") }; if (d.TryGetProperty("stop_sequence", out var ss) && ss.ValueKind != JsonValueKind.Null) debugInfo["stop_sequence"] = ss.GetString(); streamErrors.Add(debugInfo); } } } if (evt.TryGetProperty("usage", out var u2) && u2.TryGetProperty("output_tokens", out var ot)) usage.OutputTokens += ot.GetInt64(); break; } case "error": { // API Error-Event (z.B. overloaded_error) var errorInfo = new Dictionary { ["event"] = "error", ["timestamp"] = DateTime.UtcNow.ToString("o"), ["raw"] = data }; if (evt.TryGetProperty("error", out var err)) { if (err.TryGetProperty("type", out var errType)) errorInfo["error_type"] = errType.GetString(); if (err.TryGetProperty("message", out var errMsg)) errorInfo["error_message"] = errMsg.GetString(); } streamErrors.Add(errorInfo); break; } } } if (currentTextBlock.Length > 0) assistantContent.Add(new { type = "text", text = currentTextBlock.ToString() }); // Stream-Fehler als JSON serialisieren (null wenn keine Fehler) string streamErrorInfo = streamErrors.Count > 0 ? JsonSerializer.Serialize(streamErrors) : null; return (stopReason, assistantContent, executedTools, streamErrorInfo); } private HttpClient CreateClient() { var client = _httpFactory.CreateClient(); client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", _token); client.DefaultRequestHeaders.Add("anthropic-version", "2023-06-01"); client.DefaultRequestHeaders.Add("anthropic-beta", "claude-code-20250219,oauth-2025-04-20"); client.DefaultRequestHeaders.Add("User-Agent", "claude-cli/2.1.2 (external, cli)"); client.DefaultRequestHeaders.Add("x-app", "cli"); client.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("application/json")); client.DefaultRequestHeaders.Add("anthropic-dangerous-direct-browser-access", "true"); return client; } public static async Task WriteSseEvent(HttpResponse response, string eventType, object data) { var json = JsonSerializer.Serialize(data); await response.WriteAsync($"event: {eventType}\ndata: {json}\n\n"); await response.Body.FlushAsync(); } private static async Task TryWriteSse(HttpResponse response, string eventType, object data, CancellationToken ct) { try { if (!ct.IsCancellationRequested) await WriteSseEvent(response, eventType, data); } catch { // Client bereits disconnected – ignorieren } } } public record ToolResult(string Id, string Result, bool IsError); public class TokenUsage { public long InputTokens { get; set; } public long OutputTokens { get; set; } public long CacheCreationTokens { get; set; } public long CacheReadTokens { get; set; } }