Merge pull request #9 from YarikVor/main

Changed base code

Author: KSemenenko

ManagedCode.OpenAI/Attributes/JsonConverterFactory/DictionaryLookupNamingPolicy.cs

@@ -0,0 +1,14 @@
+using System.Text.Json;
+
+namespace ManagedCode.OpenAI.Chats.Enums;
+
+internal class DictionaryLookupNamingPolicy : JsonNamingPolicyDecorator 
+{
+    readonly Dictionary<string, string> dictionary;
+
+    public DictionaryLookupNamingPolicy(Dictionary<string, string> dictionary, JsonNamingPolicy underlyingNamingPolicy) 
+        : base(underlyingNamingPolicy) => this.dictionary = dictionary ?? throw new ArgumentNullException();
+    
+    public override string ConvertName (string name) 
+        => dictionary.TryGetValue(name, out var value) ? value : base.ConvertName(name);
+}

ManagedCode.OpenAI/Attributes/JsonConverterFactory/JsonNamingPolicyDecorator.cs

@@ -0,0 +1,14 @@
+using System.Text.Json;
+
+namespace ManagedCode.OpenAI.Chats.Enums;
+
+public class JsonNamingPolicyDecorator : JsonNamingPolicy 
+{
+    readonly JsonNamingPolicy underlyingNamingPolicy;
+    
+    public JsonNamingPolicyDecorator(JsonNamingPolicy underlyingNamingPolicy) 
+        => this.underlyingNamingPolicy = underlyingNamingPolicy;
+
+    public override string ConvertName (string name) => 
+        underlyingNamingPolicy?.ConvertName(name) ?? name;
+}

ManagedCode.OpenAI/Attributes/JsonConverterFactory/JsonStringEnumMemberConverter.cs

@@ -0,0 +1,47 @@
+using System.Reflection;
+using System.Runtime.Serialization;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+
+namespace ManagedCode.OpenAI.Chats.Enums;
+
+public class JsonStringEnumMemberConverter : JsonConverterFactory
+{
+    private readonly JsonNamingPolicy namingPolicy;
+    private readonly bool allowIntegerValues;
+    private readonly JsonStringEnumConverter jsonStringEnumConverter;
+
+    public JsonStringEnumMemberConverter() : this(null, true) { }
+
+    public JsonStringEnumMemberConverter(JsonNamingPolicy namingPolicy = null, bool allowIntegerValues = true)
+    {
+        this.namingPolicy = namingPolicy;
+        this.allowIntegerValues = allowIntegerValues;
+        jsonStringEnumConverter = new JsonStringEnumConverter(namingPolicy, allowIntegerValues);
+    }
+
+    public override JsonConverter CreateConverter(Type typeToConvert, JsonSerializerOptions options)
+    {
+        var query = typeToConvert
+            .GetFields(BindingFlags.Public | BindingFlags.Static)
+            .Select(field =>
+                new Tuple<string, string>(
+                    field.Name,
+                    field.GetCustomAttribute<EnumMemberAttribute>()?.Value ?? null
+                )
+            )
+            .Where(tuple => tuple.Item2 != null);
+        
+        var dictionary = query.ToDictionary(p => p.Item1, p => p.Item2);
+        if (dictionary.Count > 0)
+        {
+            return new JsonStringEnumConverter(new DictionaryLookupNamingPolicy(dictionary, namingPolicy), allowIntegerValues).CreateConverter(typeToConvert, options);
+        }
+        else
+        {
+            return jsonStringEnumConverter.CreateConverter(typeToConvert, options);
+        }
+    }
+
+    public override bool CanConvert(Type typeToConvert) => this.jsonStringEnumConverter.CanConvert(typeToConvert);
+}

ManagedCode.OpenAI/Chats/Builders/ChatRequestBuilder.cs

@@ -0,0 +1,154 @@
+using System;
+using System.Collections.Generic;
+using System.ComponentModel.DataAnnotations;
+using System.Linq;
+using System.Net.Http;
+using System.Net.Http.Json;
+using System.Text;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+using System.Threading.Tasks;
+using ManagedCode.OpenAI.Chats.Enums;
+using ManagedCode.OpenAI.Client;
+using ManagedCode.OpenAI.Exceptions;
+using ManagedCode.OpenAI.Models;
+
+namespace ManagedCode.OpenAI.Chats;
+
+public class ChatRequestBuilder
+{
+    private OpenAIClient _client;
+    private ChatRequestOptions _option;
+
+    public ChatRequestBuilder(OpenAIClient client, ChatModel model)
+    {
+        this._client = client;
+        _option = new ChatRequestOptions()
+        {
+            Model = model
+        };
+    }
+
+    public ChatRequestBuilder AskUser(string text)
+    {
+        _option.Messages.Add(new Message(RoleType.user, text));
+
+        return this;
+    }
+    
+    public ChatRequestBuilder AskAssistant(string text)
+    {
+        _option.Messages.Add(new Message(RoleType.assistant, text));
+
+        return this;
+    }
+    
+    
+
+
+    public ChatRequestBuilder WithMaxTockens(int maxTokens)
+    {
+        if (maxTokens is < 0 or > 2048)
+            throw new ArgumentOutOfRangeException();
+
+        _option.MaxTokens = maxTokens;
+
+        return this;
+    }
+
+    public ChatRequestBuilder WithTemperature(float temperature)
+    {
+        if (temperature is < 0f or > 2)
+            throw new ArgumentOutOfRangeException();
+
+        _option.Temperature = temperature;
+
+        return this;
+    }
+
+    public ChatRequestBuilder WithNucleus(float value)
+    {
+        _option.TopP = value;
+
+        return this;
+    }
+
+    public ChatRequestBuilder WithResultCount(int count)
+    {
+        if (count is < 0)
+            throw new ArgumentOutOfRangeException();
+
+        _option.N = count;
+
+        return this;
+    }
+
+    public ChatRequestBuilder WithStream()
+    {
+        _option.Stream = true;
+
+        return this;
+    }
+
+
+    public ChatRequestBuilder WithPresencePenalty(float number)
+    {
+        if (number is < -2f or > 2f)
+            throw new ArgumentOutOfRangeException();
+
+        _option.PresencePenalty = number;
+
+        return this;
+    }
+
+    public ChatRequestBuilder WithFrequencyPenalty(float number)
+    {
+        if (number is < -2f or > 2f)
+            throw new ArgumentOutOfRangeException();
+
+        _option.FrequencyPenalty = number;
+
+        return this;
+    }
+
+
+    public ChatRequestBuilder WithLogitBias(Dictionary<string, int> dictionary)
+    {
+        _option.LogitBias = dictionary;
+
+        return this;
+    }
+
+    public ChatRequestBuilder AddLogitBias(string key, int value)
+    {
+        var dict = _option.LogitBias ??= new Dictionary<string, int>();
+
+        dict.Add(key, value);
+
+        return this;
+    }
+
+    public ChatRequestBuilder WithUsername(string user)
+    {
+        _option.User = user;
+
+        return this;
+    }
+
+
+    public ChatRequestBuilder Reset()
+    {
+        _option = new ChatRequestOptions()
+        {
+            Model = _option.Model
+        };
+
+        return this;
+    }
+    
+    public ChatRequestManager Create()
+    {
+        
+        return new ChatRequestManager(_option.DeepClone(), _client);;
+    }
+}

ManagedCode.OpenAI/Chats/Builders/ChatRequestBuilderMethods.cs

@@ -0,0 +1,15 @@
+using ManagedCode.OpenAI.Chats.Enums;
+using ManagedCode.OpenAI.Client;
+using ManagedCode.OpenAI.Models;
+
+namespace ManagedCode.OpenAI.Chats;
+
+public static class ChatRequestBuilderMethods
+{
+    
+    public static ChatRequestBuilder CreateChatBuilder(this OpenAIClient client, ChatModel model)
+    {
+        return new ChatRequestBuilder(client, model);
+    }
+
+}