Fixed a bug that prevented using @Usermessage in AI Services (langcha…

…in4j#31)
AbdullahGheith · Jul 16, 2023 · 77a767a · 77a767a
1 parent 120c6a0
commit 77a767a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/langchain4j/src/main/java/dev/langchain4j/service/AiServices.java b/langchain4j/src/main/java/dev/langchain4j/service/AiServices.java
@@ -334,8 +334,9 @@ private static void validateParameters(Method method) {
         for (Parameter parameter : parameters) {
             V v = parameter.getAnnotation(V.class);
             UserMessage userMessage = parameter.getAnnotation(UserMessage.class);
-            if (v == null && userMessage == null) {
-                throw illegalConfiguration("Parameter '%s' of method '%s' should be annotated either with @V or @UserMessage", parameter.getName(), method.getName());
+            UserName userName = parameter.getAnnotation(UserName.class);
+            if (v == null && userMessage == null && userName == null) {
+                throw illegalConfiguration("Parameter '%s' of method '%s' should be annotated either with @V or @UserMessage or @UserName", parameter.getName(), method.getName());
             }
         }
     }