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