From 0d2b45ca72ad84c9b6af766e50fceef7a22adc9e Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Sat, 14 Sep 2024 15:10:04 +0800
Subject: [PATCH] bugfix

---
 rtcllm/aav.py | 6 +++---
 rtcllm/rtc.py | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/rtcllm/aav.py b/rtcllm/aav.py
index 8119179..7470aff 100644
--- a/rtcllm/aav.py
+++ b/rtcllm/aav.py
@@ -8,7 +8,7 @@ class MyMediaPlayer(MediaPlayer):
 	pass
 
 class MyTrackBase(MediaStreamTrack):
-	def __init__(self, source=None):
+	def __init__(self, source):
 		super().__init__()
 		self.dumb = None
 		self.source = source
@@ -47,13 +47,13 @@ class MyTrackBase(MediaStreamTrack):
 class MyAudioStreamTrack(MyTrackBase):
 	kind = 'audio'
 	def __init__(self, source):
-		super().__init__(source=source)
+		super().__init__(source)
 		self.dumb = AudioStreamTrack()
 		
 class MyVideoStreamTrack(MyTrackBase):
 	kind = 'video'
 	def __init__(self, source):
-		super().__init__(source=source)
+		super().__init__(source)
 		self.dumb = AudioStreamTrack()
 
 """
diff --git a/rtcllm/rtc.py b/rtcllm/rtc.py
index f94f05e..b96ccd3 100644
--- a/rtcllm/rtc.py
+++ b/rtcllm/rtc.py
@@ -51,7 +51,6 @@ RTCPeerConnection.get_local_candidates = pc_get_local_candidates
 
 class RTCLLM:
 	def __init__(self, ws_url, iceServers):
-		# self.stt_model = WhisperModel('large-v3', device="cuda", compute_type="float16")
 		self.ws_url = ws_url
 		self.omni_infer = OmniInference(ckpt_dir='/d/models/mini-omni')
 		self.iceServers = iceServers
@@ -132,6 +131,7 @@ class RTCLLM:
 		# pc.addTrack(LoopingVideoTrack('./1.mp4'))
 		# pc.addTrack(player.video)
 		await self.ws_send(json.dumps({'type':'callAccepted', 'to':data['from']}))
+		print('auto_accept_call() end')
 	
 	async def pc_track(self, peerid, track):
 		peer = self.peers[peerid]