diff --git a/cookbook/misc/clickhouse.py b/cookbook/misc/clickhouse.py index 66e2c9142..b40a9346b 100644 --- a/cookbook/misc/clickhouse.py +++ b/cookbook/misc/clickhouse.py @@ -1,55 +1,53 @@ import clickhouse_connect +import datetime as datetime +import os -# client = clickhouse_connect.get_client( -# host='hjs1z7j37j.us-east1.gcp.clickhouse.cloud', -# port=8443, -# username='default', -# password='M~PimRs~c3Z6b', -# secure=False -# ) +client = clickhouse_connect.get_client( + host=os.getenv("CLICKHOUSE_HOST"), + port=int(os.getenv("CLICKHOUSE_PORT")), + username=os.getenv("CLICKHOUSE_USERNAME"), + password=os.getenv("CLICKHOUSE_PASSWORD"), +) import clickhouse_connect -client = clickhouse_connect.get_client() - - row1 = [ - "123456", # request_id + "ishaan", # request_id "GET", # call_type "api_key_123", # api_key 50.00, # spend 1000, # total_tokens 800, # prompt_tokens 200, # completion_tokens - "2024-02-24 12:00", # startTime (replace with the actual timestamp) - "2024-02-24 13:00", # endTime (replace with the actual timestamp) + datetime.datetime.now(), # startTime (replace with the actual timestamp) + datetime.datetime.now(), # endTime (replace with the actual timestamp) "gpt-3.5", # model "user123", # user '{"key": "value"}', # metadata (replace with valid JSON) - True, # cache_hit + "True", # cache_hit "cache_key_123", # cache_key "tag1,tag2", # request_tags ] row2 = [ - "789012", # request_id + "jaffer", # request_id "POST", # call_type "api_key_456", # api_key 30.50, # spend 800, # total_tokens 600, # prompt_tokens 200, # completion_tokens - "2024-02-24 14:00", # startTime (replace with the actual timestamp) - "2024-02-24 15:00", # endTime (replace with the actual timestamp) + datetime.datetime.now(), # startTime (replace with the actual timestamp) + datetime.datetime.now(), # endTime (replace with the actual timestamp) "gpt-4.0", # model "user456", # user '{"key": "value"}', # metadata (replace with valid JSON) - False, # cache_hit + "False", # cache_hit "cache_key_789", # cache_key "tag3,tag4", # request_tags ] data = [row1, row2] -client.insert( +resp = client.insert( "spend_logs", data, column_names=[ @@ -70,3 +68,5 @@ client.insert( "request_tags", ], ) + +print(resp)