heimoshuiyu/cucyuqing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Compare commits

base: heimoshuiyu:71fd6882279ee83c9664ea10baca19e8e6365bc9
Branches Tags
heimoshuiyu:master
..
compare: heimoshuiyu:43ddd8665c28081dd641000dedb4767e80bfa36e
Branches Tags
heimoshuiyu:master

2 Commits
71fd688227 ... 43ddd8665c

Author SHA1 Message Date
heimoshuiyu
43ddd8665c fix: ES 同步接口时间精确到秒 2024-09-20 11:44:04 +08:00
heimoshuiyu
1e1f92461a typo 2024-09-19 14:35:34 +08:00
2 changed files with 4 additions and 4 deletions
Expand all files Collapse all files

6
cucyuqing/cmd/embedding.py
View File

@@ -10,14 +10,14 @@ import hashlib
from pydantic import BaseModel from pydantic import BaseModel
from typing import Any, Literal from typing import Any, Literal
from cucyuqing.pg import pool, get_cur from cucyuqing.pg import pool, get_cur
from cucyuqing.config import OPENAI_API_KEY, OPENAI_BASE_URL from cucyuqing.config import OPENAI_EMBEDDING_API_KEY, OPENAI_EMBEDDING_BASE_URL
from cucyuqing.utils import print from cucyuqing.utils import print
EmbeddingModel = Literal["acge-large-zh", "text-embedding-3-small"] EmbeddingModel = Literal["acge-large-zh", "text-embedding-3-small"]
embedding_client = openai.AsyncOpenAI( embedding_client = openai.AsyncOpenAI(
api_key=OPENAI_API_KEY, api_key=OPENAI_EMBEDDING_API_KEY,
base_url=OPENAI_BASE_URL, base_url=OPENAI_EMBEDDING_BASE_URL,
) )

2
cucyuqing/cmd/es-sync.py
View File

@@ -28,7 +28,7 @@ class ESInterval(pydantic.BaseModel):
def format_datetime(dt: datetime.datetime) -> str: def format_datetime(dt: datetime.datetime) -> str:
return dt.strftime("%Y%m%d%H") return dt.strftime("%Y%m%d%H%M%S")
def parse_unixtime(unixtime: int) -> datetime.datetime: def parse_unixtime(unixtime: int) -> datetime.datetime: