main.go 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. package main
  2. import (
  3. "data_ai/ai"
  4. "data_ai/extract"
  5. "data_ai/tool"
  6. "data_ai/udp"
  7. "data_ai/ul"
  8. "fmt"
  9. log "github.com/donnie4w/go-logger/logger"
  10. "net/http"
  11. )
  12. func init() {
  13. ul.IsLocal = false //是否本地
  14. ul.IsService = false //是否服务
  15. ul.IsTool = true //是否工具
  16. ul.IsFull = false //是否全量
  17. ul.IsFlash = true //是否flash-线上增量开关
  18. if ul.IsTool {
  19. ul.InitToolVar()
  20. } else {
  21. if ul.IsService {
  22. tool.InitCleanDeepSeekService()
  23. go http.ListenAndServe(":12321", nil)
  24. return
  25. }
  26. ul.InitGlobalVar()
  27. if !ul.IsFull {
  28. if ul.IsFlash {
  29. udp.InitProcessVar()
  30. } else {
  31. if ul.ModelType == "deepseek" {
  32. go extract.RunDeepSeek()
  33. }
  34. }
  35. }
  36. }
  37. log.Debug("测试一下转换MD方法···")
  38. detail := `<table border=\"1\"><tbody><tr><th>采购人名称</th><td>云南中医药大学</td></tr><tr><th>中标(成交)供应商名称</th><td>云南智蓝云鸽信息科技有限公司</td></tr><tr><th>合同金额</th><td>224.99万元 人民币</td></tr><tr><th>合同期限</th><td>年</td></tr><tr><th>合同签署时间</th><td>2023-12-29 00:00:00</td></tr></tbody></table>`
  39. new_detail := ul.HttpConvertToMarkdown(detail)
  40. log.Debug(new_detail)
  41. }
  42. func main() {
  43. if ul.IsTool {
  44. tool.StartToolInfo()
  45. return
  46. }
  47. if ul.IsFull {
  48. extract.ExtractFullInfo("6776b8000000000000000000")
  49. return
  50. }
  51. lock := make(chan bool)
  52. <-lock
  53. }
  54. func ttt() {
  55. var pmt = `
  56. 你是一个文本处理模型,根据下述信息做"项目分类":
  57. "项目分类"参考资料:
  58. 1、水-第三方综合服务:[地表水精细化管理服务、河流水质保障第三方服务];
  59. 2、水-监测、检测及运维服务:[排口监测、水质监测站运维、农村污水设施运维、水源地水质监测、断面监测];
  60. 3、水-设备销售:[排口规范化建设、水质监测设备];
  61. 4、水-智慧环保平台:[排污口监管系统、管网系统、水质监测系统];
  62. 5、水-工程建设:[水源地整治工程、管网建设、农村污水治理工程、水环境工程、水生态工程、雨水管网工程、排口整治];
  63. 6、水-专项咨询服务:[排口排查溯源、水源综合整治、农村污水治理、美丽河湖方案、污水排水综合规划、水生态技术服务、地下水污染防治技术服务、一河一策方案编制、排污许可];
  64. 7、大气-第三方综合服务:[驻场服务、精细化、大气污染第三方服务、空气质量达标管控服务、抑尘服务];
  65. 8、大气-监测、检测及运维服务:[网格监测、普通监测、VOCs检测、移动排放监测、机动车排放监测、监测设备运维、异味监测、空气自动监测站运维、尾气监测、加油站油气排放监测、餐饮油烟监测];
  66. 9、大气-设备销售:[监测能力建设、监测能力提升];
  67. 10、大气-智慧环保平台:[监控系统平台、大气管理系统];
  68. 11、大气-专项咨询服务:[污染源评估、污染源溯源、排放清单、重污染天气应急预案、大气污染技术咨询、秸秆焚烧治理、温室气体融合清单、异味治理、大气污染源解析、一点一策];
  69. 12、固废-监测、检测及运维服务:[垃圾填埋场监测];
  70. 13、固废-智慧环保平台:[固废相关系统平台];
  71. 14、固废-专项咨询服务:[矿渣治理、无废城市建设评价、固废污染源整治设计、固废技术审查、固废治理效果评估];
  72. 15、土壤-监测、检测及运维服务:[土壤监测、地块环境监测];
  73. 16、土壤-设备销售:[土壤监测设备];
  74. 17、土壤-智慧环保平台:[土壤监测相关系统平台];
  75. 18、土壤-专项咨询服务:[土壤污染状况风险评估、土壤修复评估、土壤金属污染综合治理、土壤重金属污染溯源];
  76. 19、噪声-监测、检测及运维服务:[声环境监测、噪声监测、交通环境噪声监测];
  77. 20、噪声-设备销售:[噪声监测设备];
  78. 21、噪声-智慧环保平台:[噪声监测平台];
  79. 22、噪声-专项咨询服务:[声环境功能区调整划分方案编制、噪声现状分析研究];
  80. 23、其他-第三方综合服务:[生态环境局第三方驻场服务、园区环保管家、园区污染第三方治理服务];
  81. 24、其他-监测、检测及运维服务:[生物多样性监测、生态环境质量监测、农村环境监测];
  82. 25、其他-设备销售:[智慧园区、环境治理系统];
  83. 26、其他-智慧环保平台:[园区环保系统平台];
  84. 27、其他-专项咨询服务:[农村环境整治、工业园区环境整治、清洁生产审核、两山基地实施方案、环评、双碳];
  85. 要求:根据正文内容,仅从"项目分类"中选取一个合适的分类。在识别过程中出现多个结果时,选取一个最贴切的,并将项目分类识别结果严格按照下述JSON格式输出。
  86. JSON格式:
  87. {
  88. "项目分类":"项目分类",
  89. }
  90. 按照以上要求输出,不要联想,不要无中生有,不要生成解释,对于尚未确定或未明确的信息请在JSON对应的值填写:无
  91. 正文内容:` + `%s` + "\n" + `
  92. 结果JSON:`
  93. title := "剑鱼园区的噪声监测系统"
  94. content := fmt.Sprintf(pmt, title)
  95. data := ai.PostZhiPuAI(content)
  96. log.Debug(data)
  97. }