Commit 5cd1cae
Fix MyPy type errors for aembedding call_type (BerriAI#16360)
Add "aembedding" to Literal type hints in ProxyLogging methods:
- pre_call_hook overloads (lines 872, 893, 913)
- during_call_hook (line 1052)
- _process_guardrail_callback (line 803)
Add type: ignore comments where ProxyLogging calls CustomLogger
callbacks (lines 1021, 1106) to handle type mismatch between
ProxyLogging's broader Literal (includes "aembedding") and
CustomLogger's narrower Literal (doesn't include "aembedding").
Related to PR BerriAI#16328 which changed embeddings endpoint to use
call_type="aembedding" for async operations.1 parent f6ddfeb commit 5cd1cae
1 file changed
+7
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
800 | 800 | | |
801 | 801 | | |
802 | 802 | | |
| 803 | + | |
803 | 804 | | |
804 | 805 | | |
805 | 806 | | |
| |||
869 | 870 | | |
870 | 871 | | |
871 | 872 | | |
| 873 | + | |
872 | 874 | | |
873 | 875 | | |
874 | 876 | | |
| |||
889 | 891 | | |
890 | 892 | | |
891 | 893 | | |
| 894 | + | |
892 | 895 | | |
893 | 896 | | |
894 | 897 | | |
| |||
908 | 911 | | |
909 | 912 | | |
910 | 913 | | |
| 914 | + | |
911 | 915 | | |
912 | 916 | | |
913 | 917 | | |
| |||
1014 | 1018 | | |
1015 | 1019 | | |
1016 | 1020 | | |
1017 | | - | |
| 1021 | + | |
1018 | 1022 | | |
1019 | 1023 | | |
1020 | 1024 | | |
| |||
1046 | 1050 | | |
1047 | 1051 | | |
1048 | 1052 | | |
| 1053 | + | |
1049 | 1054 | | |
1050 | 1055 | | |
1051 | 1056 | | |
| |||
1098 | 1103 | | |
1099 | 1104 | | |
1100 | 1105 | | |
1101 | | - | |
| 1106 | + | |
1102 | 1107 | | |
1103 | 1108 | | |
1104 | 1109 | | |
| |||
0 commit comments