build fix

tryAGI · Nov 4, 2023 · ff3cf0f · ff3cf0f
1 parent 6acef7b
commit ff3cf0f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/tests/LangChain.Core.UnitTests/Chains/RetrievalQa/RetrievalQaChainTests.cs b/src/tests/LangChain.Core.UnitTests/Chains/RetrievalQa/RetrievalQaChainTests.cs
@@ -28,7 +28,7 @@ public async Task Retrieval_Ok()
                 m => m.GetRelevantDocumentsAsync(
                     It.Is<string>(x => x == "question?"),
                     It.IsAny<string>(),
-                    It.IsAny<Callbacks?>(),
+                    It.IsAny<ICallbacks?>(),
                     It.IsAny<bool>(),
                     It.IsAny<List<string>>(),
                     It.IsAny<Dictionary<string, object>>()),
@@ -52,11 +52,11 @@ private Mock<BaseRetriever> CreateRetrieverMock()
                 .GetRelevantDocumentsAsync(
                     It.IsAny<string>(),
                     It.IsAny<string>(),
-                    It.IsAny<Callbacks>(),
+                    It.IsAny<ICallbacks>(),
                     It.IsAny<bool>(),
                     It.IsAny<List<string>>(),
                     It.IsAny<Dictionary<string, object>>()))
-            .Returns<string, string, Callbacks, bool, List<string>, Dictionary<string, object>>((query, _, _, _, _, _) =>
+            .Returns<string, string, ICallbacks, bool, List<string>, Dictionary<string, object>>((query, _, _, _, _, _) =>
             {
                 var docs = new List<Document>
                 {