Skip to content

Commit f580b7d

Browse files
Re-update type
1 parent 8695e65 commit f580b7d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

guardrails/applications/text2sql.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from string import Template
55
from typing import Callable, Dict, Optional, Type, cast
66

7-
from guardrails.classes import OT, ValidationOutcome
7+
from guardrails.classes import ValidationOutcome
88
from guardrails.document_store import DocumentStoreBase, EphemeralDocumentStore
99
from guardrails.embedding import EmbeddingBase, OpenAIEmbedding
1010
from guardrails.guard import Guard
@@ -200,7 +200,7 @@ def __call__(self, text: str) -> Optional[str]:
200200
},
201201
**self.llm_api_kwargs,
202202
)
203-
response = cast(ValidationOutcome[OT], response)
203+
response = cast(ValidationOutcome, response)
204204
validated_output: Dict = cast(Dict, response.validated_output)
205205
output = validated_output["generated_sql"]
206206
except TypeError:

0 commit comments

Comments
 (0)