intentkit 0.6.4__py3-none-any.whl → 0.6.5.dev2__py3-none-any.whl

intentkit/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.6.4"
+__version__ = "0.6.5-dev2"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

intentkit/config/config.py

@@ -51,11 +51,11 @@ class Config:
             }
         else:
             self.db = {
-                "username": os.getenv("DB_USERNAME"),
-                "password": os.getenv("DB_PASSWORD"),
-                "host": os.getenv("DB_HOST"),
-                "port": os.getenv("DB_PORT"),
-                "dbname": os.getenv("DB_NAME"),
+                "username": self.load("DB_USERNAME"),
+                "password": self.load("DB_PASSWORD"),
+                "host": self.load("DB_HOST"),
+                "port": self.load("DB_PORT"),
+                "dbname": self.load("DB_NAME"),
             }
         # ==== this part can be load from env or aws secrets manager
         self.db["auto_migrate"] = self.load("DB_AUTO_MIGRATE", "true") == "true"
@@ -66,6 +66,7 @@ class Config:
         # Redis
         self.redis_host = self.load("REDIS_HOST")
         self.redis_port = int(self.load("REDIS_PORT", "6379"))
+        self.redis_db = int(self.load("REDIS_DB", "0"))
         # AWS
         self.aws_s3_bucket = self.load("AWS_S3_BUCKET")
         self.aws_s3_cdn_url = self.load("AWS_S3_CDN_URL")

intentkit/models/db.py

@@ -38,8 +38,11 @@ async def init_db(
     # Initialize psycopg pool and AsyncPostgresSaver if not already initialized
     if _langgraph_checkpointer is None:
         if host:
+            conn_string = (
+                f"postgresql://{username}:{quote_plus(password)}@{host}:{port}/{dbname}"
+            )
             pool = AsyncConnectionPool(
-                conninfo=f"postgresql://{username}:{quote_plus(password)}@{host}:{port}/{dbname}",
+                conninfo=conn_string,
                 min_size=3,
                 max_size=20,
                 timeout=60,
@@ -47,7 +50,9 @@ async def init_db(
             )
             _langgraph_checkpointer = AsyncPostgresSaver(pool)
             if auto_migrate:
-                await _langgraph_checkpointer.setup()
+                # Migrate can not use pool, so we start from scratch
+                async with AsyncPostgresSaver.from_conn_string(conn_string) as saver:
+                    await saver.setup()
         else:
             _langgraph_checkpointer = InMemorySaver()
     # Initialize SQLAlchemy engine with pool settings

{intentkit-0.6.4.dist-info → intentkit-0.6.5.dev2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.4
+Version: 0.6.5.dev2
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestal-network/intentkit
 Project-URL: Repository, https://github.com/crestal-network/intentkit

{intentkit-0.6.4.dist-info → intentkit-0.6.5.dev2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-intentkit/__init__.py,sha256=IyBjtVSbzPGdo40SOCLABGRa2x-0vp9MP1Pxweaf_7I,378
+intentkit/__init__.py,sha256=Ktjqvztx9kMpPWpXXjP8M1TFg7t_Zvvkt3-rIViSt0w,383
 intentkit/abstracts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 intentkit/abstracts/agent.py,sha256=108gb5W8Q1Sy4G55F2_ZFv2-_CnY76qrBtpIr0Oxxqk,1489
 intentkit/abstracts/api.py,sha256=ZUc24vaQvQVbbjznx7bV0lbbQxdQPfEV8ZxM2R6wZWo,166
@@ -11,7 +11,7 @@ intentkit/clients/__init__.py,sha256=sQ_6_bRC2MPWLPH-skQ3qsEe8ce-dUGL7i8VJOautHg
 intentkit/clients/cdp.py,sha256=_CkvnBkzdq7-sFMGct4lz85FpaOoHxOGstWubhClzrA,5921
 intentkit/clients/twitter.py,sha256=gT7lXYAeRACZGYrTOPjoGoVNWAC1-qn9gkASVTeDL4M,19106
 intentkit/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-intentkit/config/config.py,sha256=6RreVvQH1xuHVOnIJ3AcaRYzdMw1RLo0vYYtvPKvTds,7453
+intentkit/config/config.py,sha256=Gn3KXgFyh4u0zmb-Awpu4AxvDFaGDa_5GrFrKBbOAXk,7509
 intentkit/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 intentkit/core/agent.py,sha256=PTHsFV_EbsfPUcTI0ErkebWurjxVykP9ISbNUsqV44Q,7764
 intentkit/core/api.py,sha256=3GIMJpwduLUSbVPNW6mQVxZncYHH3OlLwdiqFtYtEAE,1208
@@ -29,7 +29,7 @@ intentkit/models/base.py,sha256=o-zRjVrak-f5Jokdvj8BjLm8gcC3yYiYMCTLegwT2lA,185
 intentkit/models/chat.py,sha256=2P9dPWEti9f7XbO6L6Kp89lcYuyymSwqzxhL4GfBmBc,18019
 intentkit/models/conversation.py,sha256=nrbDIw-3GK5BYi_xkI15FLdx4a6SNrFK8wfAGLCsrqk,9032
 intentkit/models/credit.py,sha256=95buGlZwIuoSPasZcqYEKvdrnUiUe2RzJCR3a2rvQvU,42397
-intentkit/models/db.py,sha256=2OpdTjQWUM9FkDP8Ni0mGLeVJ5q9ah3bGlGe9-IzDp0,3999
+intentkit/models/db.py,sha256=Ked8-PsvKvqxgvytGpWlDxZNgEuyac-uGM0xWWshKmY,4210
 intentkit/models/db_mig.py,sha256=vT6Tanm-BHC2T7dTztuB1UG494EFBAlHADKsNzR6xaQ,3577
 intentkit/models/generator.py,sha256=lyZu9U9rZUGkqd_QT5SAhay9DY358JJY8EhDSpN8I1M,10298
 intentkit/models/llm.py,sha256=f_F11cjeXACu5qDB1JaCltWqS_uVb8GqOSmPTq1LYo8,25123
@@ -390,7 +390,7 @@ intentkit/utils/random.py,sha256=DymMxu9g0kuQLgJUqalvgksnIeLdS-v0aRk5nQU0mLI,452
 intentkit/utils/s3.py,sha256=9trQNkKQ5VgxWsewVsV8Y0q_pXzGRvsCYP8xauyUYkg,8549
 intentkit/utils/slack_alert.py,sha256=s7UpRgyzLW7Pbmt8cKzTJgMA9bm4EP-1rQ5KXayHu6E,2264
 intentkit/utils/tx.py,sha256=2yLLGuhvfBEY5n_GJ8wmIWLCzn0FsYKv5kRNzw_sLUI,1454
-intentkit-0.6.4.dist-info/METADATA,sha256=9bQnQqP-hBDk7vY2WV_mMkE9rHN5ZdHSaTV0ZQOfHf8,6316
-intentkit-0.6.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-intentkit-0.6.4.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
-intentkit-0.6.4.dist-info/RECORD,,
+intentkit-0.6.5.dev2.dist-info/METADATA,sha256=U12Mck8IEQssx59fvyIs_WQ37GAICSVCxfLe77Ge9Vw,6321
+intentkit-0.6.5.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+intentkit-0.6.5.dev2.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
+intentkit-0.6.5.dev2.dist-info/RECORD,,