openai/openai-dotnet
Publicmirrored from https://github.com/openai/openai-dotnetAvailable
codegen/generator/src/Visitors/ExplicitConversionFromClientResultVisitor.cs
26lines · modecode
| 1 | using Microsoft.TypeSpec.Generator.ClientModel; |
| 2 | using Microsoft.TypeSpec.Generator.Primitives; |
| 3 | using Microsoft.TypeSpec.Generator.Providers; |
| 4 | using System.ClientModel; |
| 5 | |
| 6 | namespace OpenAILibraryPlugin.Visitors; |
| 7 | |
| 8 | /// <summary> |
| 9 | /// This visitor removes the generated implicit operator that converts model to BinaryContent. |
| 10 | /// </summary> |
| 11 | public class ExplicitConversionFromClientResultVisitor : ScmLibraryVisitor |
| 12 | { |
| 13 | protected override MethodProvider VisitMethod(MethodProvider method) |
| 14 | { |
| 15 | if (method.Signature.Modifiers.HasFlag(MethodSignatureModifiers.Explicit) && |
| 16 | method.Signature.Modifiers.HasFlag(MethodSignatureModifiers.Operator) && |
| 17 | method.Signature.Parameters.Count == 1 && |
| 18 | method.Signature.Parameters[0].Type.Name == nameof(ClientResult) && |
| 19 | !method.EnclosingType.DeclarationModifiers.HasFlag(TypeSignatureModifiers.Internal)) |
| 20 | { |
| 21 | return null; |
| 22 | } |
| 23 | |
| 24 | return method; |
| 25 | } |
| 26 | } |
| 27 | |