Compare commits

...

2 Commits

Author SHA1 Message Date
Steve Nyemba 4114b25b42 bug fix
2 weeks ago
Steve Nyemba ca75f16bdc bug fix
2 weeks ago

@ -14,7 +14,7 @@
"llm": "/home/steve/dev/data/qcms/azure-openai.json", "llm": "/home/steve/dev/data/qcms/azure-openai.json",
"catalogs":{ "catalogs":{
"duckdb":{"sql":"SELECT table_name, TO_JSON(ARRAY_AGG(column_name)) as columns FROM INFORMATION_SCHEMA.COLUMNS GROUP BY table_name"}, "duckdb":{"sql":"SELECT table_name, TO_JSON(ARRAY_AGG(column_name)) as columns FROM INFORMATION_SCHEMA.COLUMNS GROUP BY table_name"},
"bigquery":{"sql":"SELECT table_name, column_name, data_type FROM :dataset.R2024Q3R4.INFORMATION_SCHEMA.COLUMNS WHERE REGEXP_CONTAINS(table_name,'^(cb_|_.+|.*_map.*)$') IS FALSE AND REGEXP_CONTAINS(column_name ,'^.*(_id|_value|_code)$') ORDER BY table_name","args":["dataset"]}, "bigquery":{"sql":"SELECT table_name, column_name, data_type FROM :dataset.INFORMATION_SCHEMA.COLUMNS WHERE REGEXP_CONTAINS(table_name,'^(cb_|_.+|.*_map.*)$') IS FALSE AND REGEXP_CONTAINS(column_name ,'^.*(_id|_value|_code)$') ORDER BY table_name","args":["dataset"]},
"postgresql":{"sql":"select table_name, column_name,data_type from information_schema.columns where table_schema = 'public' order by table_name"}, "postgresql":{"sql":"select table_name, column_name,data_type from information_schema.columns where table_schema = 'public' order by table_name"},
"sqlite":{"sql":"select tbl_name as table_name, json_group_array(y.name) as columns from sqlite_master x INNER JOIN PRAGMA_TABLE_INFO(tbl_name) y group by table_name'}"} "sqlite":{"sql":"select tbl_name as table_name, json_group_array(y.name) as columns from sqlite_master x INNER JOIN PRAGMA_TABLE_INFO(tbl_name) y group by table_name'}"}
} }

@ -127,14 +127,13 @@ def AIProxy (_label,_query,_path,_CATALOG) :
_kwargs = json.loads((open(_path)).read()) _kwargs = json.loads((open(_path)).read())
_agent = Agent(**_kwargs) _agent = Agent(**_kwargs)
_qcat = _CATALOG[_provider] _qcat = _CATALOG[_provider]
# if 'args' in _CATALOG[_provider] : if 'args' in _CATALOG[_provider] :
# _entry = transport.registry.get(_label) _entry = transport.registry.get(_label)
# for _var in _CATALOG[_provider]['args'] : for _var in _CATALOG[_provider]['args'] :
# if _var in _entry: if _var in _entry:
# _value = _entry[_var] _value = _entry[_var]
# _about = f'{_about}\n{_var} = {_value}' _about = f'{_about}\n{_var} = {_value}'
# _qcat['sql'] = _qcat['sql'].replace(f':{_var}',_value) _qcat['sql'] = _qcat['sql'].replace(f':{_var}',_value)
_catalog = _qreader.read(**_qcat).to_csv(index=0) _catalog = _qreader.read(**_qcat).to_csv(index=0)
_about = f"The queries will run on {_provider} database.\n{_about}" _about = f"The queries will run on {_provider} database.\n{_about}"

Loading…
Cancel
Save