mirror of
https://github.com/barelyprofessional/KfChatDotNet.git
synced 2026-05-02 04:22:04 -04:00
Add !hostess command (#12)
* Add basic !hostess command * Add openrouter integration
This commit is contained in:
@@ -39,6 +39,7 @@ public class BotServices
|
||||
private PeerTube? _peerTubeStatusCheck;
|
||||
private Owncast? _owncastStatusCheck;
|
||||
private ShuffleDotUs? _shuffleDotUs;
|
||||
public OpenRouter? OpenRouter;
|
||||
|
||||
private Task? _websocketWatchdog;
|
||||
private Task? _howlggGetUserTimer;
|
||||
@@ -89,7 +90,8 @@ public class BotServices
|
||||
BuildDLiveStatusCheck(),
|
||||
BuildPeerTubeLiveStatusCheck(),
|
||||
BuildOwncastLiveStatusCheck(),
|
||||
BuildShuffleDotUs()
|
||||
BuildShuffleDotUs(),
|
||||
BuildOpenRouter()
|
||||
];
|
||||
try
|
||||
{
|
||||
@@ -105,6 +107,13 @@ public class BotServices
|
||||
_websocketWatchdog = WebsocketWatchdog();
|
||||
_howlggGetUserTimer = HowlggGetUserTimer();
|
||||
}
|
||||
|
||||
private async Task BuildOpenRouter()
|
||||
{
|
||||
_logger.Debug("Building OpenRouter client");
|
||||
var proxySetting = await SettingsProvider.GetValueAsync(BuiltIn.Keys.Proxy);
|
||||
OpenRouter = new OpenRouter(proxySetting.Value);
|
||||
}
|
||||
|
||||
private async Task BuildShuffle()
|
||||
{
|
||||
|
||||
105
KfChatDotNetBot/Services/OpenRouter.cs
Normal file
105
KfChatDotNetBot/Services/OpenRouter.cs
Normal file
@@ -0,0 +1,105 @@
|
||||
using System.Net;
|
||||
using NLog;
|
||||
using System.Text.Json.Serialization;
|
||||
using System.Text.Json;
|
||||
using System.Net.Http.Headers;
|
||||
using System.Text;
|
||||
using KfChatDotNetBot.Settings;
|
||||
|
||||
public class OpenRouter(string? proxy = null)
|
||||
{
|
||||
|
||||
class OrResponse
|
||||
{
|
||||
[JsonPropertyName("id")] public string Id { get; set; } = string.Empty;
|
||||
[JsonPropertyName("provider")] public string Provider { get; set; } = string.Empty;
|
||||
[JsonPropertyName("model")] public string Model { get; set; } = string.Empty;
|
||||
[JsonPropertyName("object")] public string Object { get; set; } = string.Empty;
|
||||
[JsonPropertyName("created")] public long Created { get; set; }
|
||||
[JsonPropertyName("choices")] public List<OrChoice> Choices { get; set; } = new List<OrChoice>();
|
||||
[JsonPropertyName("usage")] public Usage Usage { get; set; } = new Usage();
|
||||
}
|
||||
|
||||
class OrChoice
|
||||
{
|
||||
[JsonPropertyName("logprobs")] public object? Logprobs { get; set; }
|
||||
[JsonPropertyName("finish_reason")] public string FinishReason { get; set; } = string.Empty;
|
||||
[JsonPropertyName("native_finish_reason")] public string NativeFinishReason { get; set; } = string.Empty;
|
||||
[JsonPropertyName("index")] public int Index { get; set; }
|
||||
[JsonPropertyName("message")] public OrMessage Message { get; set; } = new OrMessage();
|
||||
}
|
||||
|
||||
class OrMessage
|
||||
{
|
||||
[JsonPropertyName("role")] public string Role { get; set; } = string.Empty;
|
||||
[JsonPropertyName("content")] public string Content { get; set; } = string.Empty;
|
||||
[JsonPropertyName("refusal")] public object? Refusal { get; set; }
|
||||
[JsonPropertyName("reasoning")] public object? Reasoning { get; set; }
|
||||
}
|
||||
|
||||
class Usage
|
||||
{
|
||||
[JsonPropertyName("prompt_tokens")] public int PromptTokens { get; set; }
|
||||
[JsonPropertyName("completion_tokens")] public int CompletionTokens { get; set; }
|
||||
[JsonPropertyName("total_tokens")] public int TotalTokens { get; set; }
|
||||
}
|
||||
|
||||
|
||||
private Logger _logger = LogManager.GetCurrentClassLogger();
|
||||
private Uri _orEndpoint = new Uri("https://openrouter.ai/api/v1/chat/completions");
|
||||
private string? _proxy = proxy;
|
||||
|
||||
public async Task<string?> GetResponseAsync(string prompt, string question, string model = "openrouter-gpt4-1106", float Temperature = 0.7f)
|
||||
{
|
||||
_logger.Info("Sending request to OpenRouter");
|
||||
var handler = new HttpClientHandler { AutomaticDecompression = DecompressionMethods.All };
|
||||
if (_proxy != null)
|
||||
{
|
||||
handler.UseProxy = true;
|
||||
handler.Proxy = new WebProxy(_proxy);
|
||||
_logger.Debug($"Configured to use proxy {_proxy}");
|
||||
}
|
||||
using var client = new HttpClient(handler);
|
||||
|
||||
try
|
||||
{
|
||||
List<(string role, string content)> msg = [("system", prompt), ("user", question)];
|
||||
var keySetting = await SettingsProvider.GetValueAsync(BuiltIn.Keys.OpenrouterApiKey);
|
||||
if (keySetting == null || string.IsNullOrWhiteSpace(keySetting.Value))
|
||||
{
|
||||
_logger.Error("OpenRouter API key is not set in settings.");
|
||||
return null;
|
||||
}
|
||||
client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", keySetting.Value);
|
||||
|
||||
var payload = new
|
||||
{
|
||||
model,
|
||||
temperature = Temperature,
|
||||
messages = msg.ConvertAll(m => new { m.role, m.content })
|
||||
};
|
||||
var json = JsonSerializer.Serialize(payload);
|
||||
var content = new StringContent(json, Encoding.UTF8, "application/json");
|
||||
|
||||
|
||||
var response = await client.PostAsync("https://openrouter.ai/api/v1/chat/completions", content);
|
||||
response.EnsureSuccessStatusCode();
|
||||
var responseBody = await response.Content.ReadAsStringAsync();
|
||||
var responseData = JsonSerializer.Deserialize<OrResponse>(responseBody);
|
||||
|
||||
if (responseData == null || responseData.Choices.Count == 0)
|
||||
{
|
||||
_logger.Error("No response from OpenRouter.");
|
||||
return null;
|
||||
}
|
||||
return responseData.Choices[0].Message.Content;
|
||||
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
_logger.Error(ex, "Error while communicating with OpenRouter.");
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user