fix: correct completion of argument types for RF7

Author: d-biehl

packages/language_server/src/robotcode/language_server/robotframework/diagnostics/library_doc.py

@@ -263,7 +263,6 @@ class TypeDoc:
     usages: List[str] = field(default_factory=list)
     members: Optional[List[EnumMember]] = None
     items: Optional[List[TypedDictItem]] = None
-
     libname: Optional[str] = None
     libtype: Optional[str] = None
 
@@ -391,6 +390,8 @@ def from_robot(arg: Any) -> ArgumentInfo:
             types=robot_arg.types_reprs
             if get_robot_version() < (7, 0)
             else [str(robot_arg.type)]
+            if not robot_arg.type.is_union
+            else [str(t) for t in robot_arg.type.nested]
             if robot_arg.type
             else None,
             kind=KeywordArgumentKind[robot_arg.kind],
@@ -1824,7 +1825,10 @@ def _get_type_docs(keywords: List[Any], custom_converters: List[Any]) -> Set[Rob
                                 kw.type_docs[arg.name] = {}
                                 for type_info in _yield_type_info(arg.type):
                                     if type_info.type is not None:
-                                        type_doc = RobotTypeDoc.for_type(type_info.type, custom_converters)
+                                        if get_robot_version() < (7, 0):
+                                            type_doc = RobotTypeDoc.for_type(type_info.type, custom_converters)
+                                        else:
+                                            type_doc = RobotTypeDoc.for_type(type_info, custom_converters)
                                         if type_doc:
                                             kw.type_docs[arg.name][type_info.name] = type_doc.name
                                             type_docs.setdefault(type_doc, set()).add(kw.name)