Skip to content

Commit 276b21a

Browse files
committed
Adding additional parameters
1 parent ac54712 commit 276b21a

File tree

14 files changed

+45
-10
lines changed

14 files changed

+45
-10
lines changed

Examples/Examples/Chat/ChatExampleOpenAi.cs

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
using Examples.Utils;
22
using MaIN.Core.Hub;
3+
using MaIN.Domain.Configuration.BackendInferenceParams;
34
using MaIN.Domain.Models.Concrete;
45

56
namespace Examples.Chat;
@@ -15,6 +16,14 @@ public async Task Start()
1516
await AIHub.Chat()
1617
.WithModel<Gpt5Nano>()
1718
.WithMessage("What do you consider to be the greatest invention in history?")
19+
.WithInferenceParams(new OpenAiInferenceParams // We could override some inference params
20+
{
21+
ResponseFormat = "text",
22+
AdditionalParams = new Dictionary<string, object>
23+
{
24+
["max_completion_tokens"] = 2137
25+
}
26+
})
1827
.CompleteAsync(interactive: true);
1928
}
2029
}

src/MaIN.Domain/Configuration/BackendInferenceParams/AnthropicInferenceParams.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
32
using Grammar = MaIN.Domain.Models.Grammar;
43

@@ -13,4 +12,5 @@ public class AnthropicInferenceParams : IBackendInferenceParams
1312
public int? TopK { get; init; }
1413
public float? TopP { get; init; }
1514
public Grammar? Grammar { get; set; }
15+
public Dictionary<string, object>? AdditionalParams { get; init; }
1616
}

src/MaIN.Domain/Configuration/BackendInferenceParams/BackendParamsFactory.cs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,4 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
3-
using MaIN.Domain.Configuration.BackendInferenceParams;
42

53
namespace MaIN.Domain.Configuration.BackendInferenceParams;
64

src/MaIN.Domain/Configuration/BackendInferenceParams/DeepSeekInferenceParams.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
32
using Grammar = MaIN.Domain.Models.Grammar;
43

@@ -15,4 +14,5 @@ public class DeepSeekInferenceParams : IBackendInferenceParams
1514
public float? PresencePenalty { get; init; }
1615
public string? ResponseFormat { get; init; }
1716
public Grammar? Grammar { get; set; }
17+
public Dictionary<string, object>? AdditionalParams { get; init; }
1818
}

src/MaIN.Domain/Configuration/BackendInferenceParams/GeminiInferenceParams.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
32
using Grammar = MaIN.Domain.Models.Grammar;
43

@@ -13,4 +12,5 @@ public class GeminiInferenceParams : IBackendInferenceParams
1312
public float? TopP { get; init; }
1413
public string[]? StopSequences { get; init; }
1514
public Grammar? Grammar { get; set; }
15+
public Dictionary<string, object>? AdditionalParams { get; init; }
1616
}

src/MaIN.Domain/Configuration/BackendInferenceParams/GroqCloudInferenceParams.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
32
using Grammar = MaIN.Domain.Models.Grammar;
43

@@ -14,4 +13,5 @@ public class GroqCloudInferenceParams : IBackendInferenceParams
1413
public float? FrequencyPenalty { get; init; }
1514
public string? ResponseFormat { get; init; }
1615
public Grammar? Grammar { get; set; }
16+
public Dictionary<string, object>? AdditionalParams { get; init; }
1717
}

src/MaIN.Domain/Configuration/BackendInferenceParams/OllamaInferenceParams.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
32
using Grammar = MaIN.Domain.Models.Grammar;
43

@@ -15,4 +14,5 @@ public class OllamaInferenceParams : IBackendInferenceParams
1514
public int? NumCtx { get; init; }
1615
public int? NumGpu { get; init; }
1716
public Grammar? Grammar { get; set; }
17+
public Dictionary<string, object>? AdditionalParams { get; init; }
1818
}

src/MaIN.Domain/Configuration/BackendInferenceParams/OpenAiInferenceParams.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
32
using Grammar = MaIN.Domain.Models.Grammar;
43

@@ -15,4 +14,5 @@ public class OpenAiInferenceParams : IBackendInferenceParams
1514
public float? PresencePenalty { get; init; }
1615
public string? ResponseFormat { get; init; }
1716
public Grammar? Grammar { get; set; }
17+
public Dictionary<string, object>? AdditionalParams { get; init; }
1818
}

src/MaIN.Domain/Configuration/BackendInferenceParams/XaiInferenceParams.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
using MaIN.Domain.Configuration;
21
using MaIN.Domain.Entities;
32
using Grammar = MaIN.Domain.Models.Grammar;
43

@@ -14,4 +13,5 @@ public class XaiInferenceParams : IBackendInferenceParams
1413
public float? FrequencyPenalty { get; init; }
1514
public float? PresencePenalty { get; init; }
1615
public Grammar? Grammar { get; set; }
16+
public Dictionary<string, object>? AdditionalParams { get; init; }
1717
}

src/MaIN.Domain/Entities/IBackendInferenceParams.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,5 @@ namespace MaIN.Domain.Entities;
55
public interface IBackendInferenceParams
66
{
77
BackendType Backend { get; }
8+
Dictionary<string, object>? AdditionalParams { get; }
89
}

0 commit comments

Comments
 (0)