fix: Fixed errors and warnings.

tryAGI · May 15, 2024 · 1bf22ec · 1bf22ec
1 parent 7163a30
commit 1bf22ec
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 11 deletions.
diff --git a/src/Providers/Anthropic/libs/Anthropic/AnthropicModel.cs b/src/Providers/Anthropic/libs/Anthropic/AnthropicModel.cs
@@ -37,9 +37,9 @@ private static Message ToMessage(MessageResponse message)
     {
         switch (message.Role)
         {
-            case "user":
+            case RoleType.User:
                 return new Message(string.Join("\r\n", message.Content.Select(s => s is TextContent textContent ? textContent.Text : string.Empty)), MessageRole.Human);
-            case "assistant":
+            case RoleType.Assistant:
                 return new Message(string.Join("\r\n", message.Content.Select(s => s is TextContent textContent ? textContent.Text : string.Empty)), MessageRole.Ai);
         }
 

diff --git a/src/Providers/LLamaSharp/src/LLamaSharpModelBase.cs b/src/Providers/LLamaSharp/src/LLamaSharpModelBase.cs
@@ -57,7 +57,7 @@ protected LLamaSharpModelBase(LLamaSharpConfiguration configuration)
             Threads = configuration.Threads,
             BatchThreads = configuration.BatchThreads,
             BatchSize = configuration.BatchSize,
-            EmbeddingMode = configuration.EmbeddingMode
+            Embeddings = configuration.EmbeddingMode,
         };
         Model = LLamaWeights.LoadFromFile(Parameters);
         Configuration = configuration;

diff --git a/src/Providers/OpenRouter/src/OpenRouterModelProvider.cs b/src/Providers/OpenRouter/src/OpenRouterModelProvider.cs
@@ -164,9 +164,6 @@ public static ChatModels GetModelById(OpenRouterModelIds modelId)
     [CLSCompliant(false)]
     public static ChatModels GetModelById(string modelId)
     {
-        var model = Models.Values.FirstOrDefault(s => s.Id == modelId);
-        if (model == null)
-            throw new KeyNotFoundException($"Model with ID {modelId} not found.");
-        return model;
+        return Models.Values.First(s => s.Id == modelId);
     }
 }
diff --git a/...r/tools/LangChain.Providers.OpenRouter.CodeGenerator/Resources/OpenRouterModelProvider.cs b/...r/tools/LangChain.Providers.OpenRouter.CodeGenerator/Resources/OpenRouterModelProvider.cs
@@ -26,9 +26,6 @@ public static ChatModels GetModelById(OpenRouterModelIds modelId)
     [CLSCompliant(false)]
     public static ChatModels GetModelById(string modelId)
     {
-        var model = Models.Values.FirstOrDefault(s => s.Id == modelId);
-        if (model == null)
-            throw new KeyNotFoundException($"Model with ID {modelId} not found.");
-        return model;
+        return Models.Values.First(s => s.Id == modelId);
     }
 }