tdoehmen commited on
Commit
6da1916
1 Parent(s): f0b0c7e

fix pydantic

Browse files
Files changed (2) hide show
  1. duckdb-nsql/eval/schema.py +2 -1
  2. requirements.txt +1 -1
duckdb-nsql/eval/schema.py CHANGED
@@ -3,6 +3,7 @@ import enum
3
 
4
  from manifest.response import Usage
5
  from pydantic.v1 import BaseModel
 
6
 
7
  DEFAULT_TABLE_NAME: str = "db_table"
8
 
@@ -111,5 +112,5 @@ class TextToSQLModelResponse(BaseModel):
111
  output: str
112
  final_prompt: str | list[dict]
113
  raw_output: str
114
- usage: Usage
115
  metadata: str | None = None
 
3
 
4
  from manifest.response import Usage
5
  from pydantic.v1 import BaseModel
6
+ from typing import Any, List, Sequence, TypeVar
7
 
8
  DEFAULT_TABLE_NAME: str = "db_table"
9
 
 
112
  output: str
113
  final_prompt: str | list[dict]
114
  raw_output: str
115
+ usage: Any
116
  metadata: str | None = None
requirements.txt CHANGED
@@ -21,7 +21,7 @@ packaging==23.2
21
  ninja==1.11.1.1
22
  packaging
23
  #./duckdb-nsql/manifest
24
- manifest-ml @ git+https://github.com/tdoehmen/manifest@till/added_openrouter
25
  flask
26
  langchain
27
  diffusers
 
21
  ninja==1.11.1.1
22
  packaging
23
  #./duckdb-nsql/manifest
24
+ manifest-ml @ git+https://github.com/tdoehmen/manifest@till/added_openrouter_pydantic_fix
25
  flask
26
  langchain
27
  diffusers